diff --git a/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.py_in b/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.py_in
index 441c155..deda8f6 100644
--- a/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.py_in
+++ b/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.py_in
@@ -42,7 +42,6 @@
 from utilities.utilities import add_postfix
 from utilities.utilities import is_platform_gp6_or_up
 from utilities.utilities import unique_string
-from utilities.utilities import rotate
 from utilities.utilities import madlib_version
 from utilities.utilities import is_platform_pg
 from utilities.utilities import get_seg_number
@@ -257,8 +256,8 @@
         # Ordered list of sql representations of each mst_key,
         #  including NULL's.  This will be used to pass the mst keys
         #  to the db as a sql ARRAY[]
-        self.all_mst_keys = [ str(mst['mst_key']) if mst else 'NULL'\
-                for mst in self.msts_for_schedule ]
+        self.all_mst_keys = [ str(mst['mst_key']) if mst else 'NULL' \
+                              for mst in self.msts_for_schedule ]
 
         # List of all dist_keys, including any extra dist keys beyond
         #  the # segments we'll be training on--these represent the