diff --git a/src/python/nimbusml/tests/preprocessing/missing_values/test_data_with_missing.py b/src/python/nimbusml/tests/preprocessing/missing_values/test_data_with_missing.py
index fb0bdc79..0dc85f6e 100644
--- a/src/python/nimbusml/tests/preprocessing/missing_values/test_data_with_missing.py
+++ b/src/python/nimbusml/tests/preprocessing/missing_values/test_data_with_missing.py
@@ -98,7 +98,8 @@ def test_input_conversion_to_float(self):
         assert_equal(result.loc[2, 'f5'], True)
         result.loc[2, 'f5'] = False
         result = ~result
-        self.assertTrue(result.all(axis=None))
+        for val in result.all().tolist():
+            self.assertTrue(val)
 
         # Check Filter
         xf = Filter()