diff --git a/mu_map/dataset/mock.py b/mu_map/dataset/mock.py
index 8cfd95122c21b805f574abc44a1155fd7a49ffcc..9f33b3c900f0a85b053a2e7efabedb04de4e783b 100644
--- a/mu_map/dataset/mock.py
+++ b/mu_map/dataset/mock.py
@@ -11,7 +11,7 @@ class MuMapMockDataset(MuMapDataset):
         recon, mu_map = super().__getitem__(0)
         recon = recon[:, :32, :, :]
         mu_map = mu_map[:, :32, :, :]
-        return recon, mu_map / 40206.0
+        return recon, mu_map
 
     def __len__(self):
         return self.len