diff --git a/Conceptual_Guide/Part_6-building_complex_pipelines/gui/client.py b/Conceptual_Guide/Part_6-building_complex_pipelines/gui/client.py
index cda0f856..257697b4 100644
--- a/Conceptual_Guide/Part_6-building_complex_pipelines/gui/client.py
+++ b/Conceptual_Guide/Part_6-building_complex_pipelines/gui/client.py
@@ -60,7 +60,7 @@ def generate(prompt):
 with gr.Blocks() as app:
     prompt = gr.Textbox(label="Prompt")
     submit_btn = gr.Button("Generate")
-    img_output = gr.Image().style(height=512)
+    img_output = gr.Image(height=512)
     submit_btn.click(fn=generate, inputs=prompt, outputs=img_output)
 
 app.launch(server_name="0.0.0.0")