diff --git a/reasoning_gym/arithmetic/__init__.py b/reasoning_gym/arithmetic/__init__.py
index ad31381e..05d42b6d 100644
--- a/reasoning_gym/arithmetic/__init__.py
+++ b/reasoning_gym/arithmetic/__init__.py
@@ -1,9 +1,13 @@
 from .chain_sum import ChainSum, ChainSumConfig
 from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig
+from .leg_counting import LegCountingConfig, LegCountingDataset, leg_counting_dataset
 
 __all__ = [
     "ChainSum",
     "ChainSumConfig",
     "ArithmeticDataset",
-    "ArithmeticDatasetConfig"
+    "ArithmeticDatasetConfig",
+    "LegCountingConfig",
+    "LegCountingDataset",
+    "leg_counting_dataset"
 ]