diff --git a/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py b/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py
index 2f4b417..e461ba7 100644
--- a/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py
+++ b/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py
@@ -573,7 +573,7 @@ def run_random_search(self):
     def get_best_model(self):
         best_model_path = self.best_model_path
         best_model =\
-            tensorflow.keras.models.clone_model(
+            tf.keras.models.clone_model(
                 tf.keras.models.load_model(best_model_path))
         # best_model = tf.keras.models.load_model(self.best_model_path)
         return best_model