diff --git a/test_module.py b/test_module.py
index a40c277..f646dec 100644
--- a/test_module.py
+++ b/test_module.py
@@ -4,7 +4,7 @@ import matplotlib as mpl
 
 class DataCleaningTestCase(unittest.TestCase):
     def test_data_cleaning(self):
-        actual = int(time_series_visualizer.df.count())
+        actual = int(time_series_visualizer.df.count(numeric_only=True))
         expected = 1238
         self.assertEqual(actual, expected, "Expected DataFrame count after cleaning to be 1238.")