diff --git a/tests/test_lottery_predictor_collect.py b/tests/test_lottery_predictor_collect.py
index 39edccd..2cafa36 100644
--- a/tests/test_lottery_predictor_collect.py
+++ b/tests/test_lottery_predictor_collect.py
@@ -5,10 +5,10 @@ import lottery_predictor.collect as collect
 
 
 def test_get_record_count() -> None:
-    assert isinstance(collect.get_record_count('PowerballDraw'), int)
-    assert isinstance(collect.get_record_count('MegaMillionsDraw'), int)
+    assert isinstance(collect.get_record_count(table_name='PowerballDraw'), int)
+    assert isinstance(collect.get_record_count(table_name='MegaMillionsDraw'), int)
     with pytest.raises(ValueError):
-        collect.get_record_count('SomeNonExistentTable')
+        collect.get_record_count(table_name='SomeNonExistentTable')
 
 
 def test_get_recent_draw_date() -> None: