diff --git a/src/Behat/SilverStripeExtension/Context/SilverStripeAwareContextInterface.php b/src/Behat/SilverStripeExtension/Context/SilverStripeAwareContextInterface.php
index fb0c041..5da6bb2 100644
--- a/src/Behat/SilverStripeExtension/Context/SilverStripeAwareContextInterface.php
+++ b/src/Behat/SilverStripeExtension/Context/SilverStripeAwareContextInterface.php
@@ -24,4 +24,11 @@ interface SilverStripeAwareContextInterface
      * @param String $database_name Temp database name
      */
     public function setDatabase($database_name);
-}
\ No newline at end of file
+
+    /**
+     * Marks steps as AJAX steps for special treatment
+     * 
+     * @param array $ajax_steps Array of step name parts to match
+     */
+     public function setAjaxEnabledSteps($ajax_steps);
+}