diff --git a/runtime/databricks/automl_runtime/sklearn/datetime_imputer.py b/runtime/databricks/automl_runtime/sklearn/datetime_imputer.py
index c897f428..32880cb7 100644
--- a/runtime/databricks/automl_runtime/sklearn/datetime_imputer.py
+++ b/runtime/databricks/automl_runtime/sklearn/datetime_imputer.py
@@ -46,7 +46,7 @@ def fit(self, X, y=None):
         y: not used
         """
         self.fill_values = {}
-        for col_name, col_value in X.iteritems():
+        for col_name, col_value in X.items():
             col_value = pd.to_datetime(col_value, errors="coerce")
             if self.strategy == 'mean':
                 self.fill_values[col_name] = col_value.mean(skipna=True)
diff --git a/runtime/environment.txt b/runtime/environment.txt
index a52cc682..44a96bc1 100644
--- a/runtime/environment.txt
+++ b/runtime/environment.txt
@@ -4,11 +4,11 @@
 category_encoders==2.6.0
 holidays==0.28
 hyperopt==0.2.7
-mlflow==2.0.1
+mlflow==2.9.2
 numpy==1.23.5
-pandas==1.5.3
+pandas==2.0.3
 pmdarima==2.0.3
 prophet==1.1.4
-pyarrow==8.0.0
-scikit-learn==1.1.1
+pyarrow==14.0.1
+scikit-learn==1.3.0
 wrapt==1.14.1