diff --git a/examples/community/one_step_unet.py b/examples/community/one_step_unet.py
index f3eaf1e0eb..7d34bfd831 100755
--- a/examples/community/one_step_unet.py
+++ b/examples/community/one_step_unet.py
@@ -12,7 +12,7 @@ class UnetSchedulerOneForwardPipeline(DiffusionPipeline):
 
     def __call__(self):
         image = torch.randn(
-            (1, self.unet.in_channels, self.unet.sample_size, self.unet.sample_size),
+            (1, self.unet.config.in_channels, self.unet.config.sample_size, self.unet.config.sample_size),
         )
         timestep = 1