diff --git a/dataset/dataset.py b/dataset/dataset.py
index baa79c33e76a5a364d754efb6321806dac01a8b3..446d281a8740047294fbe0c45d582d4f6d812842 100644
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@@ -11,7 +11,7 @@ class Threshold_noise:
         self.threshold = threshold
 
     def __call__(self, x):
-        return torch.where((x <= self.threshold), 0.)
+        return torch.where((x <= self.threshold), 0.,x)
 
 class Log_normalisation:
     """Log normalisation of intensities"""