diff --git a/examples/arm/aot_arm_compiler.py b/examples/arm/aot_arm_compiler.py
index c663b150fd8..384df3c077d 100644
--- a/examples/arm/aot_arm_compiler.py
+++ b/examples/arm/aot_arm_compiler.py
@@ -671,9 +671,7 @@ def save_bpte_program(exec_prog, original_model: torch.nn.Module, output_name: s
     save_bundled_program(exec_prog, method_test_suites, output_name)
 
 
-def quantize_model(
-    exported_program, args, model: torch.nn.Module, example_inputs, compile_spec
-):
+def quantize_model(args, model: torch.nn.Module, example_inputs, compile_spec):
     model_int8 = quantize(
         model,
         args.model_name,
@@ -705,7 +703,7 @@ def to_edge_TOSA_delegate(
     model_int8 = None
     if args.quantize:
         model_int8, exported_program = quantize_model(
-            exported_program, args, model, example_inputs, compile_spec
+            args, model, example_inputs, compile_spec
         )
         model = model_int8
 
@@ -741,7 +739,7 @@ def to_edge_no_delegate(exported_program, args, model: torch.nn.Module, example_
             args.memory_mode,
         )
         model, exported_program = quantize_model(
-            exported_program, args, model, example_inputs, compile_spec
+            args, model, example_inputs, compile_spec
         )
         model_int8 = model