diff --git a/backend/tests/test_robust_boundary_file_size.py b/backend/tests/test_robust_boundary_file_size.py
index 263f31c..325b4d1 100644
--- a/backend/tests/test_robust_boundary_file_size.py
+++ b/backend/tests/test_robust_boundary_file_size.py
@@ -1,6 +1,8 @@
 """
 TC_003 - Robust boundary test for file size above maximum size
 """
+from django.test import TestCase
+
 def test_file_size_robust_boundary(self):
   # Creates a file with invalid size
   invalid_above_file = SimpleUploadedFile("test_above_max.txt", b"x" * (max_size + 1))