diff --git a/llms/mlx_lm/tuner/datasets.py b/llms/mlx_lm/tuner/datasets.py
index cdda7abd..7bae2862 100644
--- a/llms/mlx_lm/tuner/datasets.py
+++ b/llms/mlx_lm/tuner/datasets.py
@@ -97,7 +97,7 @@ class CompletionsDatasetCollection:
 
         while True:
             try:
-                if (curr_idx + 1) < len(item):
+                if (curr_idx + 1) <= len(item):
                     return item[curr_idx]
                 else:
                     curr_idx -= len(item)