diff --git a/README.md b/README.md
index 6ec42671..f457b8a4 100644
--- a/README.md
+++ b/README.md
@@ -101,7 +101,7 @@ See the [Dataset Gallery](https://github.com/open-thought/reasoning-gym/blob/mai
 - `WordLadderDataset`: Generate word ladder puzzles where one word is transformed into another by changing one letter at a time
 - `GroupAnagramsDataset`: Group anagrams together in a list of words
 - `IsomorphicStrings`: Check if two strings are isomorphic (have the same character mapping)
-- `RotateMatrix`: Rotate a matrix by X degrees clockwise
+- `RotateMatrix`: Rotate a matrix K times by 90 degrees clockwise
 
 ### <small>Code Tasks</small>
 
diff --git a/reasoning_gym/algorithmic/rotate_matrix.py b/reasoning_gym/algorithmic/rotate_matrix.py
index 799a23a9..ac50a281 100644
--- a/reasoning_gym/algorithmic/rotate_matrix.py
+++ b/reasoning_gym/algorithmic/rotate_matrix.py
@@ -61,7 +61,7 @@ class RotateMatrixDataset(ProceduralDataset):
         return matrix
 
     def _get_rotated(self, matrix: list[list[int]], num_rotations: int) -> list[list[int]]:
-        """Rotate the matrix by 90 degrees clockwise"""
+        """Rotate the matrix K times by 90 degrees clockwise"""
         num_rotations %= 4
         n = len(matrix)
         output = deepcopy(matrix)