diff --git a/tests/acceptance/config/behat-core.yml b/tests/acceptance/config/behat-core.yml
index dd0e9a1aeb..3d4ba8e27a 100644
--- a/tests/acceptance/config/behat-core.yml
+++ b/tests/acceptance/config/behat-core.yml
@@ -21,6 +21,7 @@ default:
         - ChecksumContext:
         - FilesVersionsContext:
         - TrashbinContext:
+        - CapabilitiesContext:
 
     coreApiAuth:
       paths:
diff --git a/tests/acceptance/features/bootstrap/CapabilitiesContext.php b/tests/acceptance/features/bootstrap/CapabilitiesContext.php
index 5bc3cd88ac..497d68b518 100644
--- a/tests/acceptance/features/bootstrap/CapabilitiesContext.php
+++ b/tests/acceptance/features/bootstrap/CapabilitiesContext.php
@@ -25,6 +25,7 @@
 use Behat\Behat\Context\Context;
 use Behat\Behat\Hook\Scope\BeforeScenarioScope;
 use PHPUnit\Framework\Assert;
+use TestHelpers\OcsApiHelper;
 
 require_once 'bootstrap.php';