diff --git a/kedro-airflow/features/steps/cli_steps.py b/kedro-airflow/features/steps/cli_steps.py
index 479c07770..759e70f2d 100644
--- a/kedro-airflow/features/steps/cli_steps.py
+++ b/kedro-airflow/features/steps/cli_steps.py
@@ -12,7 +12,7 @@
 def init_airflow(context, home_dir):
     context.airflow_dir = context.temp_dir / home_dir
     context.env["AIRFLOW_HOME"] = str(context.airflow_dir)
-    res = run([context.airflow, "db", "init"], env=context.env)
+    res = run(["airflow", "db", "migrate"], env=context.env)
     assert res.returncode == 0
 
 
diff --git a/kedro-airflow/pyproject.toml b/kedro-airflow/pyproject.toml
index d537aa7d4..309252162 100644
--- a/kedro-airflow/pyproject.toml
+++ b/kedro-airflow/pyproject.toml
@@ -29,6 +29,7 @@ test = [
     "behave",
     "black~=22.0",
     "connexion<3.0.0", # TODO: Temporary fix, connexion has changed their API, but airflow hasn't caught up yet
+    "Flask-Session<0.6", # TODO: Temporary pin, 0.6 breaks apache-airflow
     "kedro-datasets",
     "pre-commit>=2.9.2",
     "pytest",