diff --git a/reasoning_gym/logic/aiw.py b/reasoning_gym/logic/aiw.py
index 2ce3a13b..ebeb50a7 100644
--- a/reasoning_gym/logic/aiw.py
+++ b/reasoning_gym/logic/aiw.py
@@ -14,13 +14,6 @@ class TaskType(Enum):
     COLLEAGUES = "colleagues"  # Added colleagues task
 
 
-class OutputFormat(Enum):
-    """Defines the output format for the generated questions."""
-    PLAIN = "plain"
-    RESTRICTED = "restricted"
-    THINKING = "thinking"
-
-
 @dataclass
 class AliceInWonderlandConfig:
     """Configuration options for the Alice in Wonderland dataset.
diff --git a/tests/test_aiw.py b/tests/test_aiw.py
index 279fcc2c..bf556cbe 100644
--- a/tests/test_aiw.py
+++ b/tests/test_aiw.py
@@ -1,6 +1,6 @@
 import pytest
 
-from reasoning_gym.logic.aiw import AliceInWonderlandConfig, AliceInWonderlandDataset, TaskType, OutputFormat
+from reasoning_gym.logic.aiw import AliceInWonderlandConfig, AliceInWonderlandDataset, TaskType
 
 
 def test_aiw_config_validation():