diff --git a/server/entertainment_decider/models.py b/server/entertainment_decider/models.py
index d28be95..0194b69 100644
--- a/server/entertainment_decider/models.py
+++ b/server/entertainment_decider/models.py
@@ -23,11 +23,15 @@ from typing import (
 import magic
 import requests
 from pony import orm
-from pony.orm.core import Query
+from pony.orm.core import Query as PonyQuery
 
 db = orm.Database()
 
 
+T = TypeVar("T")
+Query = Union[List[T], PonyQuery]
+
+
 THUMBNAIL_ALLOWED_TYPES = [
     "image/avif",
     "image/jpeg",