diff --git a/src/ports/postgres/modules/deep_learning/madlib_keras_predict.py_in b/src/ports/postgres/modules/deep_learning/madlib_keras_predict.py_in
index d23d765..3c7b28e 100644
--- a/src/ports/postgres/modules/deep_learning/madlib_keras_predict.py_in
+++ b/src/ports/postgres/modules/deep_learning/madlib_keras_predict.py_in
@@ -377,7 +377,6 @@
         # Since the test data isn't mini-batched,
         # we have to make sure that the test data np array has the same
         # number of dimensions as input_shape. So we add a dimension to x.
-
         independent_var_filtered = []
         for i in independent_var:
             if i is not None:
@@ -389,7 +388,7 @@
         # and not mini-batched, this list contains exactly one list in it,
         # so return back the first list in probs.
         result = []
-        if len(independent_var_filtered) > 1:
+        if len(probs) > 1:
             for i in probs:
                 for j in i[0]:
                     result.append(j)