diff --git a/tests/test_python.py b/tests/test_python.py
index f15dd48e..8fbab54c 100644
--- a/tests/test_python.py
+++ b/tests/test_python.py
@@ -202,7 +202,7 @@ def test_workflow():
     model.train(data="coco8.yaml", epochs=1, imgsz=32, optimizer="SGD")
     model.val(imgsz=32)
     model.predict(SOURCE, imgsz=32)
-    model.export(format="torchscript")
+    model.export(format="onnx")
 
 
 def test_predict_callback_and_setup():