diff --git a/run_CNN.py b/run_CNN.py
index 3f5c2b56a35c635a6a49f18058efc64742c1bc4b..4b25fe538611f9ac269f48d953d71ae0ec341edf 100644
--- a/run_CNN.py
+++ b/run_CNN.py
@@ -50,7 +50,7 @@ class Dataset(torch.utils.data.Dataset):
             return None
         sig = sig[:, args.channel]
         if fs != models.get[args.specie]['fs']:
-            sig = signal.resample(sig, args.lensample * models.get[args.specie]['fs'])
+            sig = signal.resample(sig, int(args.lensample * models.get[args.specie]['fs']))
         sig = norm(sig)
         return torch.tensor(sig).float(), sample