diff --git a/src/caked/dataloader.py b/src/caked/dataloader.py
index 95ab944..ee68c95 100644
--- a/src/caked/dataloader.py
+++ b/src/caked/dataloader.py
@@ -166,7 +166,7 @@ def process(self, paths: list[str], datatype: str):
         for i in transforms:
             if i.startswith("rescale"):
                 transforms.remove(i)
-                rescale = int(i.split("=")[-1])
+                rescale = int(float(i.split("=")[-1]))
 
         if len(transforms) > 0:
             msg = f"The following transformations are not supported: {transforms}"