diff --git a/denoising_diffusion_pytorch/classifier_free_guidance.py b/denoising_diffusion_pytorch/classifier_free_guidance.py
index 6e403c7ff..c0b88236c 100644
--- a/denoising_diffusion_pytorch/classifier_free_guidance.py
+++ b/denoising_diffusion_pytorch/classifier_free_guidance.py
@@ -646,7 +646,7 @@ def model_predictions(self, x, t, classes, cond_scale = 6., rescaled_phi = 0.7,
         if self.objective == 'pred_noise':
             pred_noise = model_output if not self.use_cfg_plus_plus else model_output_null
 
-            x_start = self.predict_start_from_noise(x, t, pred_noise)
+            x_start = self.predict_start_from_noise(x, t, model_output)
             x_start = maybe_clip(x_start)
 
         elif self.objective == 'pred_x0':
diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py
index a33997dd1..55fa725bd 100644
--- a/denoising_diffusion_pytorch/version.py
+++ b/denoising_diffusion_pytorch/version.py
@@ -1 +1 @@
-__version__ = '2.1.0'
+__version__ = '2.1.1'