diff --git a/ingestion/operators/docker/Dockerfile b/ingestion/operators/docker/Dockerfile
index c165d8d02e9..23dd2f21414 100644
--- a/ingestion/operators/docker/Dockerfile
+++ b/ingestion/operators/docker/Dockerfile
@@ -49,6 +49,7 @@ COPY ingestion/operators/docker/main.py .
 RUN pip install --upgrade pip
 
 ARG INGESTION_DEPENDENCY="all"
+RUN pip install --upgrade openmetadata-ingestion[airflow]
 RUN pip install --upgrade openmetadata-ingestion[${INGESTION_DEPENDENCY}]
 # Uninstalling psycopg2-binary and installing psycopg2 instead
 # because the psycopg2-binary generates a architecture specific error
diff --git a/ingestion/operators/docker/Dockerfile-dev b/ingestion/operators/docker/Dockerfile-dev
index 8224c32c11e..fb58c76eeae 100644
--- a/ingestion/operators/docker/Dockerfile-dev
+++ b/ingestion/operators/docker/Dockerfile-dev
@@ -54,6 +54,7 @@ COPY ingestion/operators/docker/main.py .
 RUN pip install --upgrade pip
 
 ARG INGESTION_DEPENDENCY="all"
+RUN pip install --upgrade ".[airflow]"
 RUN pip install --upgrade ".[${INGESTION_DEPENDENCY}]"
 
 # Uninstalling psycopg2-binary and installing psycopg2 instead