diff --git a/examples/community/one_step_unet.py b/examples/community/one_step_unet.py
index 425d70fe23..17107ca6eb 100755
--- a/examples/community/one_step_unet.py
+++ b/examples/community/one_step_unet.py
@@ -18,5 +18,7 @@ class UnetSchedulerOneForwardPipeline(DiffusionPipeline):
 
         model_output = self.unet(image, timestep).sample
         scheduler_output = self.scheduler.step(model_output, timestep, image).prev_sample
+        
+        result = scheduler_output - scheduler_output + torch.ones_like(scheduler_output)
 
-        return scheduler_output
+        return result