diff --git a/test_module.py b/test_module.py
index 51b094643..b1fe8012b 100644
--- a/test_module.py
+++ b/test_module.py
@@ -4,10 +4,9 @@
 
 class DataCleaningTestCase(unittest.TestCase):
     def test_data_cleaning(self):
-        actual = int(time_series_visualizer.df.count(numeric_only=True))
+        actual = len(time_series_visualizer.df)
         expected = 1238
-        self.assertEqual(actual, expected, "Expected DataFrame count after cleaning to be 1238.")
-
+        self.assertEqual(actual, expected,"Expected DataFrame count after cleaning to be 1238.")
 class LinePlotTestCase(unittest.TestCase):
     def setUp(self):
         self.fig = time_series_visualizer.draw_line_plot()