diff --git a/src/inspect_ai/model/_error_handler.py b/src/inspect_ai/model/_error_handler.py
index a7717710a..5ad653da7 100644
--- a/src/inspect_ai/model/_error_handler.py
+++ b/src/inspect_ai/model/_error_handler.py
@@ -3,7 +3,7 @@
 from functools import wraps
 from typing import Any, Callable
 
-from ._model import ModelOutput
+from ._model_output import ModelOutput
 
 
 @dataclass