diff --git a/src/Tests/BlockClassTest.php b/src/Tests/BlockClassTest.php
index 071d634..f14a261 100644
--- a/src/Tests/BlockClassTest.php
+++ b/src/Tests/BlockClassTest.php
@@ -1,8 +1,4 @@
 <?php
-/**
- * @file
- * Contains \Drupal\block_class\BlockClassTest.
- */
 
 namespace Drupal\block_class\Tests;
 
@@ -39,7 +35,7 @@ class BlockClassTest extends WebTestBase {
       'region' => 'content',
       'third_party_settings[block_class][classes]' => 'TestClass_content'
     );
-    $this->drupalPostForm(NULL, $edit, t('Save block'));
+    $this->drupalPostForm(NULL, $edit, $this->t('Save block'));
 
     // Add a user account menu with a custom CSS class.
     $this->drupalGet('admin/structure/block/add/system_menu_block:account/classy', ['query' => ['region' => 'content']]);
@@ -47,7 +43,7 @@ class BlockClassTest extends WebTestBase {
       'region' => 'secondary_menu',
       'third_party_settings[block_class][classes]' => 'TestClass_menu'
     );
-    $this->drupalPostForm(NULL, $edit, t('Save block'));
+    $this->drupalPostForm(NULL, $edit, $this->t('Save block'));
 
     // Go to the front page of the user.
     $this->drupalGet('<front>');