diff --git a/src/Tests/TokenFieldUiTest.php b/src/Tests/TokenFieldUiTest.php
index 6da6256..2de8e9d 100644
--- a/src/Tests/TokenFieldUiTest.php
+++ b/src/Tests/TokenFieldUiTest.php
@@ -187,6 +187,9 @@ class TokenFieldUiTest extends TokenTestBase {
       'field_image:thumbnail:url' => $style_thumbnail->buildUrl('public://example1.png'),
       'field_image:medium:url' => $style_medium->buildUrl('public://example1.png'),
       'field_image:large:url' => $style_large->buildUrl('public://example1.png'),
+      'field_image:thumbnail' => $style_thumbnail->buildUrl('public://example1.png'),
+      'field_image:medium' => $style_medium->buildUrl('public://example1.png'),
+      'field_image:large' => $style_large->buildUrl('public://example1.png'),
       // field_image_2
       'field_image_2:thumbnail:mimetype' => 'image/gif',
       'field_image_2:medium:mimetype' => 'image/gif',
@@ -206,6 +209,9 @@ class TokenFieldUiTest extends TokenTestBase {
       'field_image_2:thumbnail:url' => $style_thumbnail->buildUrl('public://example2.gif'),
       'field_image_2:medium:url' => $style_medium->buildUrl('public://example2.gif'),
       'field_image_2:large:url' => $style_large->buildUrl('public://example2.gif'),
+      'field_image_2:thumbnail' => $style_thumbnail->buildUrl('public://example2.gif'),
+      'field_image_2:medium' => $style_medium->buildUrl('public://example2.gif'),
+      'field_image_2:large' => $style_large->buildUrl('public://example2.gif'),
     ];
     $this->assertTokens('node', ['node' => $node], $tokens);
   }