diff --git a/__init__.py b/__init__.py
index 4d974da..57939c5 100644
--- a/__init__.py
+++ b/__init__.py
@@ -19,7 +19,7 @@ bp = Blueprint(directory_name, __name__, template_folder="templates")
 
 def create_flag_if_missing(challenge_id: int, user_id: int, flag_content: str):
     user_id_str = str(user_id)
-    flag = Flags.query.filter_by(id=challenge_id, data=user_id_str).first()
+    flag = Flags.query.filter_by(challenge_id=challenge_id, data=user_id_str).first()
 
     if flag is None:
         new_flag = Flags(