diff --git a/judge/views/test_formatter/test_formatter.py b/judge/views/test_formatter/test_formatter.py
index 6eb5cb62..44314367 100644
--- a/judge/views/test_formatter/test_formatter.py
+++ b/judge/views/test_formatter/test_formatter.py
@@ -178,7 +178,7 @@ def get(self, request):
         for i in range(len(preview_file)):
             response = response + (f"<p>{preview_file[i]}</p>\n")
 
-        files_list = [preview_file[0], preview_file[1]]
+        files_list = preview_file[:2] if len(preview_file) > 2 else preview_file
 
         return render(
             request,