diff --git a/templates/problem/data.html b/templates/problem/data.html
index 8910509..dfa366a 100644
--- a/templates/problem/data.html
+++ b/templates/problem/data.html
@@ -235,7 +235,7 @@
                     if (window.valid_files[i].endsWith(".in") || window.valid_files[i].endsWith(".inp") || window.valid_files[i].endsWith(".IN") || window.valid_files[i].endsWith(".INP")) {
                         inFiles.push(window.valid_files[i]);
                     }
-                    if (window.valid_files[i].endsWith(".out") || window.valid_files[i].endsWith(".OUT")) {
+                    if (window.valid_files[i].endsWith(".out") || window.valid_files[i].endsWith(".ans") || window.valid_files[i].endsWith(".OUT") || window.valid_files[i].endsWith(".ANS")) {
                         outFiles.push(window.valid_files[i]);
                     }
                 }