diff --git a/autotranscript/audio.py b/autotranscript/audio.py
index 04feb1d..4d457b6 100644
--- a/autotranscript/audio.py
+++ b/autotranscript/audio.py
@@ -98,7 +98,10 @@ class AudioProcessor:
         """
         
         start = int(start * self.sr)
-        end = int(torch.ceil(end * self.sr))
+        if (isinstance(end, float) or isinstance(end, int)) and isinstance(self.sr, int):
+            end = int(np.ceil(end * self.sr))
+        else:
+            end = int(torch.ceil(end * self.sr))
         return self.waveform[start:end]
 
     @staticmethod