diff --git a/clip_vision.py b/clip_vision.py
index dd341df..8c18f60 100644
--- a/clip_vision.py
+++ b/clip_vision.py
@@ -68,7 +68,7 @@ class ClipVision(NNOperator):
             return False
         x = torch.randn((1, 3, 224, 224))
 
-        torch.onnx.export(self.model, x, output_file, input_names=['INPUT0'],
+        torch.onnx.export(self.model.model, x, output_file, input_names=['INPUT0'],
                           output_names=['OUTPUT0'],  dynamic_axes={'INPUT0': [0]})
         return True