diff --git a/biogtr/datasets/base_dataset.py b/biogtr/datasets/base_dataset.py
index 01753a42..15b87d45 100644
--- a/biogtr/datasets/base_dataset.py
+++ b/biogtr/datasets/base_dataset.py
@@ -29,7 +29,7 @@ def __init__(
 
         Args:
             label_files: a list of paths to label files.
-            should at least contain detections for inference, detections + tracks for training.
+                should at least contain detections for inference, detections + tracks for training.
             vid_files: list of paths to video files.
             padding: amount of padding around object crops
             crop_size: the size of the object crops