diff --git a/dal/db_query/queries.sql b/dal/db_query/queries.sql
index bfd2edf..9a13e26 100644
--- a/dal/db_query/queries.sql
+++ b/dal/db_query/queries.sql
@@ -168,7 +168,7 @@ SELECT * FROM quiz
 WHERE
     deleted = false AND
     archived = false AND
-    status = 'start' AND
+    (status = 'start' OR status = 'ai') AND
     qid = $1;
 
 -- name: GetQuestionTitle :one
diff --git a/dal/sqlcgen/queries.sql.go b/dal/sqlcgen/queries.sql.go
index 2dc6fa4..ba76f7e 100644
--- a/dal/sqlcgen/queries.sql.go
+++ b/dal/sqlcgen/queries.sql.go
@@ -2229,7 +2229,7 @@ SELECT id, qid, accountid, deleted, archived, fingerprinting, repeatable, note_p
 WHERE
     deleted = false AND
     archived = false AND
-    status = 'start' AND
+    (status = 'start' OR status = 'ai') AND
     qid = $1
 `