diff --git a/reasoning_gym/algorithmic/word_reversal.py b/reasoning_gym/algorithmic/word_reversal.py
index ce9f273d..5ad94ea7 100644
--- a/reasoning_gym/algorithmic/word_reversal.py
+++ b/reasoning_gym/algorithmic/word_reversal.py
@@ -27,7 +27,7 @@ class WordSequenceReversalConfig:
 class WordSequenceReversalDataset(ProceduralDataset):
     """Generates word sequence reversal tasks from text spans"""
 
-    def __init__(self, config: WordReversalConfig):
+    def __init__(self, config: WordSequenceReversalConfig):
         super().__init__(config=config, seed=config.seed, size=config.size)
 
         # Load and preprocess text
diff --git a/tests/test_word_reversal.py b/tests/test_word_reversal.py
index bcf042ce..15680223 100644
--- a/tests/test_word_reversal.py
+++ b/tests/test_word_reversal.py
@@ -13,7 +13,7 @@ def test_word_sequence_reversal_config_validation():
         config.validate()
 
     with pytest.raises(AssertionError):
-        config = WordReversalConfig(min_words=10, max_words=5)
+        config = WordSequenceReversalConfig(min_words=10, max_words=5)
         config.validate()