diff --git a/Makefile b/Makefile
index b4aa8e6..c9b5865 100644
--- a/Makefile
+++ b/Makefile
@@ -53,7 +53,7 @@ make-migrations:
 	@docker compose run --rm web python manage.py makemigrations
 
 migrate:
-	@docker compose run --rm web python manage.py migrate
+	@docker compose run --rm web python manage.py migrate $(migration_args)
 
 collectstatic:
 	@docker compose run --rm web python manage.py collectstatic --noinput
diff --git a/app/general/tests/test_document_admin.py b/app/general/tests/test_document_admin.py
index a3c6bf4..3b31e3d 100644
--- a/app/general/tests/test_document_admin.py
+++ b/app/general/tests/test_document_admin.py
@@ -9,7 +9,7 @@
 from general.models import Document, Institution
 
 
-class TestDocumentForm(unittest.TestCase):
+class TestDocumentForm(TestCase):
     def __init__(self, methodName: str = "runTest"):
         super().__init__(methodName)
         self.form = None