diff --git a/amt/data.py b/amt/data.py
index 98f9244..7440082 100644
--- a/amt/data.py
+++ b/amt/data.py
@@ -113,7 +113,7 @@ def get_wav_segments(
     if pad_last == True:
         yield torch.nn.functional.pad(
             buffer[stride_samples:],
-            (0.0, chunk_samples - len(buffer[stride_samples:])),
+            (0, chunk_samples - len(buffer[stride_samples:])),
         )