diff --git a/get_train_annot.py b/get_train_annot.py
index 0ebe2870bb2c16796069ce887d82836b2c44a4bd..3f556fa4b351c816f12a53b6707682aa0f143491 100755
--- a/get_train_annot.py
+++ b/get_train_annot.py
@@ -36,7 +36,7 @@ def main(entry, arguments, species_list):
         0, file_duration, arguments.duration - arguments.overlap)
     grp = utils.split_annotations(
         grp, arguments.duration)
-    grp = pd.reset_index(drop=True)
+    grp = grp.reset_index(drop=True)
     while len(grp) != 0:
         # collect all the data between the offset and duration-overlap
         if len(offset_list) >= 1: