diff --git a/examples/quantization/TARGETS b/examples/quantization/TARGETS
index 231c56567d4..7d8471ac576 100644
--- a/examples/quantization/TARGETS
+++ b/examples/quantization/TARGETS
@@ -1,6 +1,6 @@
-load("@fbcode_macros//build_defs:python_binary.bzl", "python_binary")
+load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
 
-python_binary(
+runtime.python_binary(
     name = "example",
     main_src = "example.py",
     deps = [
diff --git a/examples/quantization/example.py b/examples/quantization/example.py
index 009ccd6a486..ef6695443e3 100644
--- a/examples/quantization/example.py
+++ b/examples/quantization/example.py
@@ -133,3 +133,4 @@ def verify_xnnpack_quantizer_matching_fx_quant_model(model_name, model, example_
         )
 
     quantize(args.model_name, model, example_inputs)
+    print("finished")