diff --git a/kedro-airflow/features/environment.py b/kedro-airflow/features/environment.py
index 1db194429..0da6ac934 100644
--- a/kedro-airflow/features/environment.py
+++ b/kedro-airflow/features/environment.py
@@ -50,8 +50,7 @@ def call(cmd, print_output=False):
 
     # pip install us
     call([context.python, "-m", "pip", "install", "-U", "pip", "pip-tools"])
-    call([context.pip, "install", "-r", "test_requirements.txt"])
-    call([context.pip, "install", "."])
+    call([context.pip, "install", ".[test]"])
 
     context.temp_dir = Path(tempfile.mkdtemp())