diff --git a/behat.yml b/behat.yml
index 90494140..d6ee8622 100644
--- a/behat.yml
+++ b/behat.yml
@@ -1,4 +1,7 @@
 default:
-  paths:
-    features: features
-    bootstrap: vendor/wp-cli/wp-cli-tests/features/bootstrap
+  suites:
+    default:
+      contexts:
+        - WP_CLI\Tests\Context\FeatureContext
+      paths:
+        - features