diff --git a/flag.module b/flag.module
index 9fbd375..03b2343 100644
--- a/flag.module
+++ b/flag.module
@@ -157,7 +157,7 @@ function flag_form_alter(&$form, FormStateInterface $form_state, $form_id) {
       $form['flag'][$flag->id()] = array(
         '#type' => 'checkbox',
         '#title' => $flag->label(),
-        '#description' => $flag->getFlagLongText(),
+        '#description' => $flag->getLongText('flag'),
         '#default_value' => is_null($flagging)? NULL: 1,
         '#return_value' => 1,
         // Used by our drupalSetSummary() on vertical tabs.
@@ -579,7 +579,7 @@ function flag_flag_insert(FlagInterface $flag) {
     $action = Action::create([
       'id' => $flag_id,
       'type' => $flag->getFlaggableEntityTypeId(),
-      'label' => $flag->getFlagShortText(),
+      'label' => $flag->getShortText('flag'),
       'plugin' => 'flag_action:' . $flag->id() . '.flag',
       'configuration' => [
         'flag_id' => $flag->id(),
@@ -593,7 +593,7 @@ function flag_flag_insert(FlagInterface $flag) {
     $action = Action::create([
       'id' => $unflag_id,
       'type' => $flag->getFlaggableEntityTypeId(),
-      'label' => $flag->getUnflagShortText(),
+      'label' => $flag->getShortText('unflag'),
       'plugin' => 'flag_action:' . $flag->id() . '.unflag',
       'configuration' => [
         'flag_id' => $flag->id(),
diff --git a/src/ActionLink/ActionLinkTypeBase.php b/src/ActionLink/ActionLinkTypeBase.php
index ad2688b..19b9a05 100644
--- a/src/ActionLink/ActionLinkTypeBase.php
+++ b/src/ActionLink/ActionLinkTypeBase.php
@@ -85,7 +85,7 @@ abstract class ActionLinkTypeBase extends PluginBase implements ActionLinkTypePl
     $action = $this->getAction($flag, $entity);
     $url = $this->getUrl($action, $flag, $entity);
     $url->setOption('query', ['destination' => $this->getDestination()]);
-    $title = $action === 'unflag' ? $flag->getUnflagShortText() : $flag->getFlagShortText();
+    $title = $flag->getShortText($action);
 
     return Link::fromTextAndUrl($title, $url);
   }
@@ -106,9 +106,9 @@ abstract class ActionLinkTypeBase extends PluginBase implements ActionLinkTypePl
         '#flaggable' => $entity,
         '#action' => $action,
         '#access' => $access->isAllowed(),
-        '#title' => $action === 'unflag' ? $flag->getUnflagShortText() : $flag->getFlagShortText(),
+        '#title' =>  $flag->getShortText($action),
         '#attributes' => [
-          'title' => $action === 'unflag' ? $flag->getUnflagLongText() : $flag->getFlagLongText(),
+          'title' => $flag->getLongText($action),
         ],
       ];
       // Build the URL. It is important that bubbleable metadata is explicitly
diff --git a/src/Entity/Flag.php b/src/Entity/Flag.php
index 3175cf9..8220dad 100644
--- a/src/Entity/Flag.php
+++ b/src/Entity/Flag.php
@@ -384,15 +384,15 @@ class Flag extends ConfigEntityBundleBase implements FlagInterface {
   /**
    * {@inheritdoc}
    */
-  public function getFlagShortText() {
-    return $this->flag_short;
+  public function getShortText($action) {
+    return $action === 'unflag' ? $this->unflag_short : $this->flag_short;
   }
 
   /**
    * {@inheritdoc}
    */
-  public function getFlagLongText() {
-    return $this->flag_long;
+  public function getLongText($action) {
+    return $action === 'unflag' ? $this->unflag_long : $this->flag_long;
   }
 
   /**
@@ -419,13 +419,6 @@ class Flag extends ConfigEntityBundleBase implements FlagInterface {
   /**
    * {@inheritdoc}
    */
-  public function getUnflagLongText() {
-    return $this->unflag_long;
-  }
-
-  /**
-   * {@inheritdoc}
-   */
   public function setUnflagLongText($unflag_long) {
     $this->unflag_long = $unflag_long;
   }
@@ -447,13 +440,6 @@ class Flag extends ConfigEntityBundleBase implements FlagInterface {
   /**
    * {@inheritdoc}
    */
-  public function getUnflagShortText() {
-    return $this->unflag_short;
-  }
-
-  /**
-   * {@inheritdoc}
-   */
   public function setUnflagShortText($unflag_short) {
     $this->unflag_short = $unflag_short;
   }
diff --git a/src/FlagInterface.php b/src/FlagInterface.php
index 7ba20b4..a26eb05 100644
--- a/src/FlagInterface.php
+++ b/src/FlagInterface.php
@@ -162,18 +162,24 @@ interface FlagInterface extends ConfigEntityInterface, EntityWithPluginCollectio
   /**
    * Gets the flag short text.
    *
+   * @param string $action
+   *   The flag action, either 'flag' or 'unflag'.
+   *
    * @return string
    *   A string containing the flag short text.
    */
-  public function getFlagShortText();
+  public function getShortText($action);
 
   /**
    * Gets the flag long text.
    *
+   * @param string $action
+   *   The flag action, either 'flag' or 'unflag'.
+   *
    * @return string
    *   A string containing the flag long text.
    */
-  public function getFlagLongText();
+  public function getLongText($action);
 
   /**
    * Sets the flag long text.
@@ -200,14 +206,6 @@ interface FlagInterface extends ConfigEntityInterface, EntityWithPluginCollectio
   public function setFlagMessage($flag_message);
 
   /**
-   * Gets the unflag short text.
-   *
-   * @return string
-   *   A string containing the unflag short text.
-   */
-  public function getUnflagShortText();
-
-  /**
    * Sets the unflag short text.
    *
    * @param string $flag_short
@@ -216,14 +214,6 @@ interface FlagInterface extends ConfigEntityInterface, EntityWithPluginCollectio
   public function setUnflagShortText($flag_short);
 
   /**
-   * Gets the flag long text.
-   *
-   * @return string
-   *   A string containing the unflag long text.
-   */
-  public function getUnflagLongText();
-
-  /**
    * Sets the unflag long text.
    *
    * @param string $unflag_long
diff --git a/src/Form/FlagConfirmForm.php b/src/Form/FlagConfirmForm.php
index 3b8b15c..77a0520 100644
--- a/src/Form/FlagConfirmForm.php
+++ b/src/Form/FlagConfirmForm.php
@@ -32,7 +32,7 @@ class FlagConfirmForm extends FlagConfirmFormBase {
    * {@inheritdoc}
    */
   public function getDescription() {
-    return $this->flag->getFlagLongText();
+    return $this->flag->getLongText('flag');
   }
 
   /**
diff --git a/src/Form/FlagFormBase.php b/src/Form/FlagFormBase.php
index 21b03c2..4f0079e 100644
--- a/src/Form/FlagFormBase.php
+++ b/src/Form/FlagFormBase.php
@@ -108,7 +108,7 @@ abstract class FlagFormBase extends EntityForm {
       '#title' => $this->t('Messages'),
     ];
 
-    $flag_short = $flag->getFlagShortText();
+    $flag_short = $flag->getShortText('flag');
     $form['messages']['flag_short'] = [
       '#type' => 'textfield',
       '#title' => $this->t('Flag link text'),
@@ -120,7 +120,7 @@ abstract class FlagFormBase extends EntityForm {
     $form['messages']['flag_long'] = [
       '#type' => 'textfield',
       '#title' => $this->t('Flag link description'),
-      '#default_value' => $flag->getFlagLongText(),
+      '#default_value' => $flag->getLongText('flag'),
       '#description' => $this->t('The description of the "flag this" link. Usually displayed on mouseover.'),
     ];
 
@@ -131,7 +131,7 @@ abstract class FlagFormBase extends EntityForm {
       '#description' => $this->t('Message displayed after flagging content. If JavaScript is enabled, it will be displayed below the link. If not, it will be displayed in the message area.'),
     ];
 
-    $unflag_short = $flag->getUnflagShortText();
+    $unflag_short = $flag->getShortText('unflag');
     $form['messages']['unflag_short'] = [
       '#type' => 'textfield',
       '#title' => $this->t('Unflag link text'),
@@ -143,7 +143,7 @@ abstract class FlagFormBase extends EntityForm {
     $form['messages']['unflag_long'] = [
       '#type' => 'textfield',
       '#title' => $this->t('Unflag link description'),
-      '#default_value' => $flag->getUnflagLongText(),
+      '#default_value' => $flag->getLongText('unflag'),
       '#description' => $this->t('The description of the "unflag this" link. Usually displayed on mouseover.'),
     ];
 
diff --git a/src/Form/UnflagConfirmForm.php b/src/Form/UnflagConfirmForm.php
index 903f2b1..f91a925 100644
--- a/src/Form/UnflagConfirmForm.php
+++ b/src/Form/UnflagConfirmForm.php
@@ -30,7 +30,7 @@ class UnflagConfirmForm extends FlagConfirmFormBase {
    * {@inheritdoc}
    */
   public function getDescription() {
-    return $this->flag->getUnflagLongText();
+    return $this->flag->getLongText('unflag');
   }
 
   /**
diff --git a/src/Form/UnflaggingForm.php b/src/Form/UnflaggingForm.php
index cb9d84f..045efa4 100644
--- a/src/Form/UnflaggingForm.php
+++ b/src/Form/UnflaggingForm.php
@@ -52,7 +52,7 @@ class UnflaggingForm extends FlagConfirmFormBase {
    * {@inheritdoc}
    */
   public function getDescription() {
-    return $this->flag->getUnflagLongText();
+    return $this->flag->getLongText('unflag');
   }
 
   /**
diff --git a/src/Plugin/Derivative/EntityFlagActionDeriver.php b/src/Plugin/Derivative/EntityFlagActionDeriver.php
index 0f6d52d..f0862e0 100644
--- a/src/Plugin/Derivative/EntityFlagActionDeriver.php
+++ b/src/Plugin/Derivative/EntityFlagActionDeriver.php
@@ -47,7 +47,7 @@ class EntityFlagActionDeriver extends DeriverBase implements ContainerDeriverInt
       foreach (['flag', 'unflag'] as $action) {
         $this->derivatives[$flag_id . '.' . $action] = [
           'id' => $flag_id . '.' . $action,
-          'label' => $action === 'unflag' ? $flag->getUnflagShortText() : $flag->getFlagShortText(),
+          'label' => $flag->getShortText($action),
           'type' => $flag->getFlaggableEntityTypeId(),
         ] + $base_plugin_definition;
       }
diff --git a/src/Tests/FlagContextualLinksTest.php b/src/Tests/FlagContextualLinksTest.php
index d9d0492..8014113 100644
--- a/src/Tests/FlagContextualLinksTest.php
+++ b/src/Tests/FlagContextualLinksTest.php
@@ -152,8 +152,7 @@ class FlagContextualLinksTest extends FlagTestBase {
       'entity_id' => $node->id(),
     ));
 
-    $flag_short_text = $flag_action == 'flag' ? $flag->getFlagShortText() : $flag->getUnflagShortText();
-
+    $flag_short_text = $flag->getShortText($flag_action);
     $matches = array();
     $url_pattern =  preg_quote(base_path()) . '(' . preg_quote($flag_url->getInternalPath(), '@') . '[^"]*)';
     $found = preg_match('@' . preg_quote('<ul class="contextual-links"><li class="flag-test-label-123"><a href="') .
diff --git a/src/Tests/FlagPermissionsTest.php b/src/Tests/FlagPermissionsTest.php
index cbcf840..565a537 100644
--- a/src/Tests/FlagPermissionsTest.php
+++ b/src/Tests/FlagPermissionsTest.php
@@ -78,8 +78,8 @@ class FlagPermissionsTest extends FlagTestBase {
     // Check the full flag permission user can flag...
     $this->drupalLogin($this->fullFlagUser);
     $this->drupalGet('node/' . $this->node->id());
-    $this->assertLink($this->flag->getFlagShortText());
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->assertLink($this->flag->getShortText('flag'));
+    $this->clickLink($this->flag->getShortText('flag'));
     $this->assertResponse(200);
 
     // ...and also unflag.
@@ -90,25 +90,25 @@ class FlagPermissionsTest extends FlagTestBase {
     // Check the flag only user can flag...
     $this->drupalLogin($this->flagOnlyUser);
     $this->drupalGet('node/' . $this->node->id());
-    $this->assertLink($this->flag->getFlagShortText());
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->assertLink($this->flag->getShortText('flag'));
+    $this->clickLink($this->flag->getShortText('flag'));
     $this->assertResponse(200);
 
     // ...but not unflag.
     $this->drupalGet('node/' . $this->node->id());
     $this->assertResponse(200);
-    $this->assertNoLink($this->flag->getFlagShortText());
-    $this->assertNoLink($this->flag->getUnflagShortText());
+    $this->assertNoLink($this->flag->getShortText('flag'));
+    $this->assertNoLink($this->flag->getShortText('unflag'));
 
     // Check an unprivileged authenticated user.
     $this->drupalLogin($this->authUser);
     $this->drupalGet('node/' . $this->node->id());
-    $this->assertNoLink($this->flag->getFlagShortText());
+    $this->assertNoLink($this->flag->getShortText('flag'));
 
     // Check the anonymous user.
     $this->drupalLogout();
     $this->drupalGet('node/' . $this->node->id());
-    $this->assertNoLink($this->flag->getFlagShortText());
+    $this->assertNoLink($this->flag->getShortText('flag'));
   }
 
 }
diff --git a/src/Tests/LinkOwnershipAccessTest.php b/src/Tests/LinkOwnershipAccessTest.php
index 23fad5c..ee8409c 100644
--- a/src/Tests/LinkOwnershipAccessTest.php
+++ b/src/Tests/LinkOwnershipAccessTest.php
@@ -54,7 +54,7 @@ class LinkOwnershipAccessTest extends FlagTestBase {
 
     // Flag the node with user 1.
     $this->drupalGet($this->node->toUrl());
-    $this->clickLink($flag->getFlagShortText());
+    $this->clickLink($flag->getShortText('flag'));
     $this->assertResponse(200);
     $this->assertLink($flag->getUnflagShortText());
 
@@ -62,7 +62,7 @@ class LinkOwnershipAccessTest extends FlagTestBase {
     $user_2 = $this->drupalCreateUser();
     $this->drupalLogin($user_2);
     $this->drupalGet($this->node->toUrl());
-    $this->assertLink($flag->getFlagShortText(), 0, "A flag link is found on the page for user 2.");
+    $this->assertLink($flag->getShortText('flag'), 0, "A flag link is found on the page for user 2.");
 
   }
 
@@ -80,15 +80,15 @@ class LinkOwnershipAccessTest extends FlagTestBase {
 
     // Flag the node with user 1.
     $this->drupalGet($this->node->toUrl());
-    $this->clickLink($flag->getFlagShortText());
+    $this->clickLink($flag->getShortText('flag'));
     $this->assertResponse(200);
-    $this->assertLink($flag->getUnflagShortText());
+    $this->assertLink($flag->getShortText('unflag'));
 
     // Switch to user 2. They should see the unflag link too.
     $user_2 = $this->drupalCreateUser();
     $this->drupalLogin($user_2);
     $this->drupalGet($this->node->toUrl());
-    $this->assertLink($flag->getUnflagShortText(), 0, "The unflag link is found on the page for user 2.");
+    $this->assertLink($flag->getShortText('unflag'), 0, "The unflag link is found on the page for user 2.");
   }
 
 }
diff --git a/src/Tests/LinkTypeAjaxTest.php b/src/Tests/LinkTypeAjaxTest.php
index 92a77a6..51f863c 100644
--- a/src/Tests/LinkTypeAjaxTest.php
+++ b/src/Tests/LinkTypeAjaxTest.php
@@ -69,18 +69,18 @@ class LinkTypeAjaxTest extends FlagTestBase {
     $this->drupalGet($node_url);
 
     // Confirm the flag link exists.
-    $this->assertLink($this->flag->getFlagShortText());
+    $this->assertLink($this->flag->getShortText('flag'));
 
     // Click the flag link. This ensures that the non-JS fallback works we are
     // redirected to back to the page and the node is flagged.
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->clickLink($this->flag->getShortText('flag'));
     $this->assertUrl($node_url);
-    $this->assertLink($this->flag->getUnflagShortText());
+    $this->assertLink($this->flag->getShortText('unflag'));
 
     // Click the unflag link, repeat the check.
-    $this->clickLink($this->flag->getUnflagShortText());
+    $this->clickLink($this->flag->getShortText('unflag'));
     $this->assertUrl($node_url);
-    $this->assertLink($this->flag->getFlagShortText());
+    $this->assertLink($this->flag->getShortText('flag'));
 
     /* Assert that initially a flag action link is displayed within a wrapper.
      *
@@ -89,7 +89,7 @@ class LinkTypeAjaxTest extends FlagTestBase {
      *
      * div[contains(concat(' ',normalize-space(@class),' '),' flag ')]
      */
-    $links = $this->xpath('//div[contains(concat(" ",normalize-space(@class)," ")," flag ")]/a[normalize-space()=:label]', array(':label' => $this->flag->getFlagShortText()));
+    $links = $this->xpath('//div[contains(concat(" ",normalize-space(@class)," ")," flag ")]/a[normalize-space()=:label]', array(':label' => $this->flag->getShortText('flag')));
     // Use the same logic as clickLink() to get an AJAX response.
     $link_target = $this->getAbsoluteUrl($links[0]['href']);
     $flag_response = $this->drupalGetAjax($link_target);
@@ -104,7 +104,7 @@ class LinkTypeAjaxTest extends FlagTestBase {
     $this->assertTrue($id_class_position !== FALSE);
 
     // Assert the flag response contains a wrapped unflag action link.
-    $this->assertEqual($this->flag->getUnflagShortText(), $flag_xml_data->a->__toString());
+    $this->assertEqual($this->flag->getShortText('unflag'), $flag_xml_data->a->__toString());
 
     // From the payload extract a unflag action link href.
     // Act as if the unflag link has been clicked.
@@ -116,7 +116,7 @@ class LinkTypeAjaxTest extends FlagTestBase {
     $this->assertTrue($unflag_class_position !== FALSE);
 
     // Assert the unflag response contains a wrapped flag action link.
-    $this->assertEqual($this->flag->getFlagShortText(), $unflag_xml_data->a->__toString());
+    $this->assertEqual($this->flag->getShortText('flag'), $unflag_xml_data->a->__toString());
 
   }
 
diff --git a/src/Tests/LinkTypeConfirmFormTest.php b/src/Tests/LinkTypeConfirmFormTest.php
index 62106f5..21e2008 100644
--- a/src/Tests/LinkTypeConfirmFormTest.php
+++ b/src/Tests/LinkTypeConfirmFormTest.php
@@ -97,7 +97,7 @@ class LinkTypeConfirmFormTest extends FlagTestBase {
 
     // Click the flag link.
     $this->drupalGet('node/' . $node_id);
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->clickLink($this->flag->getShortText('flag'));
 
     // Check if we have the confirm form message displayed.
     $this->assertText($this->flagConfirmMessage);
@@ -131,7 +131,7 @@ class LinkTypeConfirmFormTest extends FlagTestBase {
 
     // Check that the node is no longer flagged.
     $this->drupalGet('node/' . $node_id);
-    $this->assertLink($this->flag->getFlagShortText());
+    $this->assertLink($this->flag->getShortText('flag'));
 
     // Check the flag count was decremented.
     $flag_count_unflagged = db_query('SELECT count FROM {flag_counts}
diff --git a/src/Tests/LinkTypeFieldEntryTest.php b/src/Tests/LinkTypeFieldEntryTest.php
index 6d11591..1f83d34 100644
--- a/src/Tests/LinkTypeFieldEntryTest.php
+++ b/src/Tests/LinkTypeFieldEntryTest.php
@@ -125,7 +125,7 @@ class LinkTypeFieldEntryTest extends FlagTestBase {
 
     // Click the flag link.
     $this->drupalGet('node/' . $this->nodeId);
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->clickLink($this->flag->getShortText('flag'));
 
     // Check if we have the confirm form message displayed.
     $this->assertText($this->flagConfirmMessage);
@@ -220,7 +220,7 @@ class LinkTypeFieldEntryTest extends FlagTestBase {
 
     // Check that the node is no longer flagged.
     $this->drupalGet('node/' . $this->nodeId);
-    $this->assertLink($this->flag->getFlagShortText());
+    $this->assertLink($this->flag->getShortText('flag'));
   }
 
 }
diff --git a/src/Tests/LinkTypeReloadTest.php b/src/Tests/LinkTypeReloadTest.php
index 470e761..4fa905a 100644
--- a/src/Tests/LinkTypeReloadTest.php
+++ b/src/Tests/LinkTypeReloadTest.php
@@ -74,7 +74,7 @@ class LinkTypeReloadTest extends FlagTestBase {
 
     // Click the flag link.
     $this->drupalGet('node/' . $node_id);
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->clickLink($this->flag->getShortText('flag'));
 
     // Check that the node is flagged.
     $this->drupalGet('node/' . $node_id);
@@ -99,7 +99,7 @@ class LinkTypeReloadTest extends FlagTestBase {
 
     // Check that the node is no longer flagged.
     $this->drupalGet('node/' . $node_id);
-    $this->assertLink($this->flag->getFlagShortText());
+    $this->assertLink($this->flag->getShortText('flag'));
 
     // Check the flag count was decremented.
     $flag_count_unflagged = db_query('SELECT count FROM {flag_counts}
diff --git a/src/Tests/UserFlagTypeTest.php b/src/Tests/UserFlagTypeTest.php
index 014c715..9286802 100644
--- a/src/Tests/UserFlagTypeTest.php
+++ b/src/Tests/UserFlagTypeTest.php
@@ -70,7 +70,7 @@ class UserFlagTypeTest extends FlagTestBase {
 
     // Assert flag appears on the profile page.
     $this->drupalGet('user/' . $user->id());
-    $this->assertLink($flag->getFlagShortText());
+    $this->assertLink($flag->getShortText('flag'));
 
     // Uncheck extra permssions.
     $edit = [
@@ -84,7 +84,7 @@ class UserFlagTypeTest extends FlagTestBase {
 
     // Assert the flag disapears from the profile page.
     $this->drupalGet('user/' . $user->id());
-    $this->assertNoLink($flag->getFlagShortText());
+    $this->assertNoLink($flag->getShortText('flag'));
   }
 
 }
diff --git a/tests/src/FunctionalJavascript/LinkTypeAjaxTest.php b/tests/src/FunctionalJavascript/LinkTypeAjaxTest.php
index eed62e7..a8c1357 100644
--- a/tests/src/FunctionalJavascript/LinkTypeAjaxTest.php
+++ b/tests/src/FunctionalJavascript/LinkTypeAjaxTest.php
@@ -88,10 +88,10 @@ class LinkTypeAjaxTest extends JavascriptTestBase {
     $this->drupalGet($this->node->toUrl());
 
     // Confirm the flag link exists.
-    $this->assertSession()->linkExists($this->flag->getFlagShortText());
+    $this->assertSession()->linkExists($this->flag->getShortText('flag'));
 
     // Click the flag link.
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->clickLink($this->flag->getShortText('flag'));
     $this->assertSession()->assertWaitOnAjaxRequest();
     $this->assertSession()->addressEquals($this->node->toUrl());
     $this->assertSession()->linkExists($this->flag->getUnflagShortText());
@@ -101,11 +101,11 @@ class LinkTypeAjaxTest extends JavascriptTestBase {
     $this->clickLink($this->flag->getUnflagShortText());
     $this->assertSession()->assertWaitOnAjaxRequest();
     $this->assertSession()->addressEquals($this->node->toUrl());
-    $this->assertSession()->linkExists($this->flag->getFlagShortText());
+    $this->assertSession()->linkExists($this->flag->getShortText('flag'));
     $this->assertFalse($this->flagService->getFlagging($this->flag, $this->node, $auth_user));
 
     // And flag again.
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->clickLink($this->flag->getShortText('flag'));
     $this->assertSession()->assertWaitOnAjaxRequest();
     $this->assertSession()->addressEquals($this->node->toUrl());
     $this->assertSession()->linkExists($this->flag->getUnflagShortText());
@@ -125,10 +125,10 @@ class LinkTypeAjaxTest extends JavascriptTestBase {
     $this->drupalGet($this->node->toUrl());
 
     // Confirm the flag link exists.
-    $this->assertSession()->linkExists($this->flag->getFlagShortText());
+    $this->assertSession()->linkExists($this->flag->getShortText('flag'));
 
     // Click the flag link.
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->clickLink($this->flag->getShortText('flag'));
     $this->assertSession()->assertWaitOnAjaxRequest();
     $this->assertSession()->addressEquals($this->node->toUrl());
     $this->assertSession()->linkExists($this->flag->getUnflagShortText());
@@ -141,14 +141,14 @@ class LinkTypeAjaxTest extends JavascriptTestBase {
     $this->clickLink($this->flag->getUnflagShortText());
     $this->assertSession()->assertWaitOnAjaxRequest();
     $this->assertSession()->addressEquals($this->node->toUrl());
-    $this->assertSession()->linkExists($this->flag->getFlagShortText());
+    $this->assertSession()->linkExists($this->flag->getShortText('flag'));
     $this->assertFalse($this->flagService->getFlagging($this->flag, $this->node, $auth_user));
 
     // Verifies that the event subscriber was called.
     $this->assertTrue($this->container->get('state')->get('flag_test.is_unflagged', FALSE));
 
     // And flag again.
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->clickLink($this->flag->getShortText('flag'));
     $this->assertSession()->assertWaitOnAjaxRequest();
     $this->assertSession()->addressEquals($this->node->toUrl());
     $this->assertSession()->linkExists($this->flag->getUnflagShortText());
diff --git a/tests/src/FunctionalJavascript/ModalFormTest.php b/tests/src/FunctionalJavascript/ModalFormTest.php
index 181520f..c6bdcbf 100644
--- a/tests/src/FunctionalJavascript/ModalFormTest.php
+++ b/tests/src/FunctionalJavascript/ModalFormTest.php
@@ -83,7 +83,7 @@ class ModalFormTest extends JavascriptTestBase {
   public function testModalOption() {
     // Verify default, non-modal behavior.
     $this->drupalGet($this->node->toUrl());
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->clickLink($this->flag->getShortText('flag'));
 
     // Should be on the confirm form page, since this isn't using a modal.
     $expected = Url::fromRoute('flag.confirm_flag', [
@@ -111,7 +111,7 @@ class ModalFormTest extends JavascriptTestBase {
     $this->flag->save();
 
     $this->drupalGet($this->node->toUrl());
-    $this->clickLink($this->flag->getFlagShortText());
+    $this->clickLink($this->flag->getShortText('flag'));
     $this->assertSession()->assertWaitOnAjaxRequest();
 
     // Should still be on the node url, as this is using a modal.
