diff --git a/emote/extra/onnx_exporter.py b/emote/extra/onnx_exporter.py
index 3bc48e06..843307d8 100644
--- a/emote/extra/onnx_exporter.py
+++ b/emote/extra/onnx_exporter.py
@@ -206,7 +206,7 @@ def _trace(self):
                 )
 
                 f.seek(0)
-                model_proto = onnx.load_model(f, onnx.ModelProto)
+                model_proto = onnx.load_model(f)
 
             return model_proto