diff --git a/examples/apps/flux_demo.py b/examples/apps/flux_demo.py
index 4e8aaf3a4e..30c6ca788d 100644
--- a/examples/apps/flux_demo.py
+++ b/examples/apps/flux_demo.py
@@ -13,7 +13,7 @@
 from diffusers.models.transformers.transformer_flux import FluxTransformer2DModel
 
 # Register SDPA as a standalone operator. Converter and lowering pass are defined in register_sdpa.py
-sys.path.append(os.path.join(os.path.dirname(__file__), "../dynamo"))
+sys.path.append(os.path.join(os.path.dirname(__file__), "../../tools/llm/torchtrt_ext"))
 from register_sdpa import *
 
 DEVICE = "cuda:0"