diff --git a/queries_to_answers.py b/queries_to_answers.py
index 80fa9ab..cd9d96f 100644
--- a/queries_to_answers.py
+++ b/queries_to_answers.py
@@ -102,12 +102,12 @@ def add_answered_flags (
 
     answer_type: QueryType
     try:
-        answer_type = QueryType (answer.query.query_type)
+        answer_type = QueryType (answer.query_ref.query_type)
     except (TypeError, ValueError):
         return
 
     if answer_type in (QueryType.YOUTUBE_COMMENT,
-                       QueryType.YOUTUBE_COMMENT,
+                       QueryType.YOUTUBE_SYSTEM,
                        QueryType.KIRIBAN,
                        QueryType.NICO_REPORT,
                        QueryType.SNACK_TIME,