diff --git a/backend/apps/library/downloads.py b/backend/apps/library/downloads.py
index 3c9752d..e9a93a7 100644
--- a/backend/apps/library/downloads.py
+++ b/backend/apps/library/downloads.py
@@ -12,7 +12,7 @@ from django.db import connection
 from django.utils import timezone
 
 from . import services
-from .models import DownloadTask
+from .models import DownloadTask, MediaItem
 
 logger = logging.getLogger(__name__)