diff --git a/core/modules/block/lib/Drupal/block/Tests/BlockTemplateSuggestionsUnitTest.php b/core/modules/block/lib/Drupal/block/Tests/BlockTemplateSuggestionsUnitTest.php
index 3b5b389..47c6014 100644
--- a/core/modules/block/lib/Drupal/block/Tests/BlockTemplateSuggestionsUnitTest.php
+++ b/core/modules/block/lib/Drupal/block/Tests/BlockTemplateSuggestionsUnitTest.php
@@ -7,13 +7,13 @@
 
 namespace Drupal\block\Tests;
 
-use Drupal\simpletest\UnitTestBase;
+use Drupal\simpletest\DrupalUnitTestBase;
 use stdClass;
 
 /**
  * Unit tests for template_preprocess_block().
  */
-class BlockTemplateSuggestionsUnitTest extends UnitTestBase {
+class BlockTemplateSuggestionsUnitTest extends DrupalUnitTestBase {
   public static function getInfo() {
     return array(
       'name' => 'Block template suggestions',
@@ -22,6 +22,11 @@ public static function getInfo() {
     );
   }
 
+  function setUp() {
+    drupal_load('module', 'block');
+    parent::setUp();
+  }
+
   /**
    * Test if template_preprocess_block() handles the suggestions right.
    */