diff --git a/core/modules/system/lib/Drupal/system/Tests/Actions/NodeTest.php b/core/modules/system/lib/Drupal/system/Tests/Actions/NodeTest.php
index 6c41ac7..438b85a 100644
--- a/core/modules/system/lib/Drupal/system/Tests/Actions/NodeTest.php
+++ b/core/modules/system/lib/Drupal/system/Tests/Actions/NodeTest.php
@@ -31,6 +31,7 @@ function testNodePropertiesActions() {
       $module_list = array('action');
       module_enable($module_list);
     }
+
     // Test the publish action.
     actions_do('node_publish_action', $node);
     $this->assertEqual($node->status, NODE_PUBLISHED, 'Node reference got updated and published.');
@@ -89,6 +90,8 @@ function testNodeSaveAction() {
 
   /**
    * Test the node unpublish by keyword action.
+   *
+   *  @see theme()
    */
   function testNodeUnpublishByKeywordAction() {
     $node = $this->_createNode();
@@ -96,6 +99,8 @@ function testNodeUnpublishByKeywordAction() {
       $module_list = array('action');
       module_enable($module_list);
     }
+    // Make sure that all modules are loaded before calling theme() function.
+    drupal_container()->get('module_handler')->loadAll();
     // Publish the node.
     $node->status = NODE_PUBLISHED;
     // Generate a pseudo-random keyword based on the title of the node.