diff --git a/ee/recommendation/ml_trainer/entrypoint.sh b/ee/recommendation/ml_trainer/entrypoint.sh
index af93b034b..b917e44c6 100755
--- a/ee/recommendation/ml_trainer/entrypoint.sh
+++ b/ee/recommendation/ml_trainer/entrypoint.sh
@@ -6,6 +6,7 @@ find airflow/ -type f -name "*.cfg" -exec sed -i "s/{{pg_port_airflow}}/${pg_por
 find airflow/ -type f -name "*.cfg" -exec sed -i "s/{{pg_dbname_airflow}}/${pg_dbname_airflow}/g" {} \;
 find airflow/ -type f -name "*.cfg" -exec sed -i "s#{{airflow_secret_key}}#${airflow_secret_key}#g" {} \;
 export MLFLOW_TRACKING_URI=postgresql+psycopg2://${pg_user_ml}:${pg_password_ml}@${pg_host_ml}:${pg_port_ml}/${pg_dbname_ml}
+mlflow db upgrade
 git init airflow/dags
 # Airflow setup
 # airflow db init