diff --git a/reasoning_gym/algorithmic/sentence_reordering.py b/reasoning_gym/algorithmic/sentence_reordering.py
index 04dfbff3..6856a264 100644
--- a/reasoning_gym/algorithmic/sentence_reordering.py
+++ b/reasoning_gym/algorithmic/sentence_reordering.py
@@ -42,6 +42,8 @@ class SentenceReorderingDataset(ProceduralDataset):
 
         Args:
             sentence (str): The correct sentence to use for dataset generation.
+            seed (int): The seed to use for random number generation.
+            idx (int): The index to add to the seed for random number generation.
             shuffle (bool): Whether to shuffle the words to create the input sentence.
 
         Returns: