diff --git a/main.py b/main.py
index caf9913..46f0c69 100644
--- a/main.py
+++ b/main.py
@@ -35,7 +35,7 @@ async def start_up():
     inspector = inspect(engine)
     if not inspector.has_table("notification"):
         # Create the Notification table
-        Base.metadata.create_all(bind=engine, tables=[Notification.__table__, NotificationSeen.__table__])
+        Base.metadata.create_all(bind=engine, tables=["notification", "notification_seen"])
         logger.info("Notification table was created.")
     else:
         logger.info("Notification table already exists.")