diff --git a/features/environment.py b/features/environment.py
index dcf47fc118..172dfd006a 100644
--- a/features/environment.py
+++ b/features/environment.py
@@ -124,6 +124,6 @@ def _install_project_requirements(context):
         .splitlines()
     )
     install_reqs = [req for req in install_reqs if "{" not in req]
-    install_reqs.append("kedro-datasets[pandas.CSVDataSet]")
+    install_reqs.append(".[pandas.CSVDataSet]")
     call([context.pip, "install", *install_reqs], env=context.env)
     return context
diff --git a/features/steps/test_starter/{{ cookiecutter.repo_name }}/src/requirements.txt b/features/steps/test_starter/{{ cookiecutter.repo_name }}/src/requirements.txt
index e18878f6db..7e6f29ac16 100644
--- a/features/steps/test_starter/{{ cookiecutter.repo_name }}/src/requirements.txt	
+++ b/features/steps/test_starter/{{ cookiecutter.repo_name }}/src/requirements.txt	
@@ -6,7 +6,7 @@ isort~=5.0
 jupyter~=1.0
 jupyterlab_server>=2.11.1, <2.16.0
 jupyterlab~=3.0, <3.6.0
-kedro-datasets[pandas.CSVDataSet]~=1.2
+kedro[pandas.CSVDataSet]=={{ cookiecutter.kedro_version }}
 kedro-telemetry~=0.2.0
 nbstripout~=0.4
 pytest-cov~=3.0