diff --git a/util/drawing.py b/util/drawing.py
index fff5e77..294d3df 100644
--- a/util/drawing.py
+++ b/util/drawing.py
@@ -8,12 +8,12 @@ import os
 from datetime import datetime, timedelta
 from dotenv import load_dotenv
 
-from data.database import PowerballDraw, MegaMillionsDraw
+from data.database import PowerballDraw, MegaMillionsDraw, Table
 
 load_dotenv()
 
 
-def check_table_name(table_name: str) -> dict | None:
+def check_table_name2(table_name: Table) -> dict | None:
     """
     :param table_name: table name
     :return: dict of game info or None