diff --git a/api/test-setup.py b/api/test-setup.py
index fa01feb1..c3955d14 100644
--- a/api/test-setup.py
+++ b/api/test-setup.py
@@ -6,6 +6,6 @@ height = 512
 width = 512
 prompt = "an astronaut eating a hamburger"
 
-pipe = OnnxStableDiffusionPipeline.from_pretrained("../../stable_diffusion_onnx", provider="DmlExecutionProvider", safety_checker=None)
+pipe = OnnxStableDiffusionPipeline.from_pretrained("../models/stable-diffusion-onnx-v1-5", provider="DmlExecutionProvider", safety_checker=None)
 image = pipe(prompt, height, width, num_inference_steps=steps, guidance_scale=cfg).images[0]
 image.save("../test.png")
\ No newline at end of file