diff --git a/hydrooj-problem-transmission/src/check.ts b/hydrooj-problem-transmission/src/check.ts
index b118eb9..a5a4476 100644
--- a/hydrooj-problem-transmission/src/check.ts
+++ b/hydrooj-problem-transmission/src/check.ts
@@ -31,7 +31,8 @@ interface JudgeConfig {
 }
 
 const ALLOW_EXTRA_TESTDATA = [
-    ...['py', 'sh', 'cc', 'cpp', 'js', 'mjs', 'ts'].map((ext: string) => `generator.${ext}`)
+    ...['py', 'sh', 'cc', 'cpp', 'js', 'mjs', 'ts']
+        .map((ext: string) => ['generator', 'validator', ''].map((name) => `${name}.${ext}`)).flat(),
 ]
 
 console.log({