diff --git a/example/files/tests/test_models.py b/example/files/tests/test_models.py
index 8fb34dd..53d459a 100644
--- a/example/files/tests/test_models.py
+++ b/example/files/tests/test_models.py
@@ -1,6 +1,8 @@
 from os import path
 
+from django.contrib.auth.models import User
 from django.test import TestCase
+from django.urls import reverse
 from files.models import CaptionedFile