diff --git a/climax/core/predict.py b/climax/core/predict.py
index 1b6ca175695f7a10ac7cec0d8483cb260b93805d..c6a93ab98fb4609fea9c7640c0bf85ffd929d262 100644
--- a/climax/core/predict.py
+++ b/climax/core/predict.py
@@ -63,13 +63,12 @@ def predict_ERA5(net, ERA5_ds, predictand, batch_size=16, **kwargs):
               'tasmin': EoDataset.add_coordinates(target[:, 1, ...].squeeze())}
     elif predictand == 'pr':
         ds = {'prob': EoDataset.add_coordinates(target[:, 0, ...].squeeze()),
-              'shape': EoDataset.add_coordinates(target[:, 1, ...].squeeze()),
-              'scale': EoDataset.add_coordinates(target[:, 2, ...].squeeze()),
 
               # amount of precipitation: expected value of gamma distribution
               # pr = shape * scale
-              'pr': EoDataset.add_coordinates((target[:, 1, ...] *
-                                               target[:, 2, ...]).squeeze())}
+              'pr': EoDataset.add_coordinates(
+                  (np.exp(target[:, 1, ...]) *
+                   np.exp(target[:, 2, ...])).squeeze())}
     else:
         # single predictand
         ds = {predictand: EoDataset.add_coordinates(target)}