diff --git a/src/Drupal/DrupalExtension/Context/DrupalContext.php b/src/Drupal/DrupalExtension/Context/DrupalContext.php
index 6528831..45da080 100644
--- a/src/Drupal/DrupalExtension/Context/DrupalContext.php
+++ b/src/Drupal/DrupalExtension/Context/DrupalContext.php
@@ -377,21 +377,17 @@ class DrupalContext extends MinkContext implements DrupalAwareInterface {
     if (!$regionObj) {
       throw new \Exception("$region region was not found");
     }
-
-    $elements = $regionObj->findAll('css', 'h2');
-    $found = FALSE;
-    if (!empty($elements)) {
-      foreach ($elements as $element) {
-        $text = $element->getText();
-        if ($text === $heading) {
-          $found = TRUE;
-          continue;
+    foreach (array('h1', 'h2', 'h3', 'h4', 'h5', 'h6') as $tag) {
+      $elements = $regionObj->findAll('css', $tag);
+      if (!empty($elements)) {
+        foreach ($elements as $element) {
+          if (trim($element->getText()) === $heading) {
+            return;
+          }
         }
       }
     }
-    if (!$found) {
-      throw new \Exception(sprintf('The heading "%s" was not found in the "%s" region.', $heading, $region));
-    }
+    throw new \Exception(sprintf('The heading "%s" was not found in the "%s" region.', $heading, $region));
   }
 
   /**