diff --git a/benchmark/run.py b/benchmark/run.py
index 2016701..9e7d783 100644
--- a/benchmark/run.py
+++ b/benchmark/run.py
@@ -134,7 +134,7 @@ elif args.format == 'onnx':
         for o_n in output_names:
             dynamic_axes[o_n] = {0: 'batch_size', 1: 'sequence_length'}
         torch.onnx.export(
-            op.model,
+            op.model.model,
             tuple(inputs.values()),
             onnx_path,
             input_names=input_names,