diff --git a/reasoning_gym/code/codeio.py b/reasoning_gym/code/codeio.py
index e992171d..d787019c 100644
--- a/reasoning_gym/code/codeio.py
+++ b/reasoning_gym/code/codeio.py
@@ -170,4 +170,4 @@ class CodeIODataset(ProceduralDataset):
 
 
 # Register the dataset
-# register_dataset("codeio", CodeIODataset, CodeIOConfig)
+register_dataset("codeio", CodeIODataset, CodeIOConfig)