diff --git a/joj3_config_generator/processers/task.py b/joj3_config_generator/processers/task.py
index 2cdc780..b8ff8a0 100644
--- a/joj3_config_generator/processers/task.py
+++ b/joj3_config_generator/processers/task.py
@@ -186,28 +186,21 @@ def fix_diff(
                 proc_limit=50,
             )
         )
-
-        # Ensure case_stage.diff and case_stage.diff.output are defined
-        diff_output = (
-            case_stage.diff.output
-            if case_stage.diff and case_stage.diff.output
-            else None
-        )
-        if diff_output:
-            parser_cases.append(
-                result.DiffCasesConfig(
-                    outputs=[
-                        result.DiffOutputConfig(
-                            score=diff_output.score,
-                            file_name="stdout",
-                            answer_path=str(base_dir / stdout),
-                            force_quit_on_diff=diff_output.force_quit,
-                            always_hide=diff_output.hide,
-                            compare_space=not diff_output.ignore_spaces,
-                        )
-                    ]
-                )
+        diff_output = case_stage.diff.output
+        parser_cases.append(
+            result.DiffCasesConfig(
+                outputs=[
+                    result.DiffOutputConfig(
+                        score=diff_output.score,
+                        file_name="stdout",
+                        answer_path=str(base_dir / stdout),
+                        force_quit_on_diff=diff_output.force_quit,
+                        always_hide=diff_output.hide,
+                        compare_space=not diff_output.ignore_spaces,
+                    )
+                ]
             )
+        )
 
     diff_executor.with_.cases = stage_cases
     diff_parser_config.with_ = result.DiffConfig(name="diff", cases=parser_cases)