diff --git a/backend/tests/test_boundary_file_size.py b/backend/tests/test_boundary_file_size.py
index 5984cdf..c8feffe 100644
--- a/backend/tests/test_boundary_file_size.py
+++ b/backend/tests/test_boundary_file_size.py
@@ -1,6 +1,8 @@
 """
 TC_002 - Boundary test for file size
 """
+from django.test import TestCase
+
 def test_file_size_boundary(self):
   # Create file with minimum valid size (1 byte)
   min_file = SimpleUploadedFile("min_file.txt", b"x" * 1)