diff --git a/metalearners/_utils.py b/metalearners/_utils.py
index ffcfbcc..7238683 100644
--- a/metalearners/_utils.py
+++ b/metalearners/_utils.py
@@ -494,7 +494,7 @@ def check_onnx_installed():
     try:
         import onnx  # noqa F401
     except ImportError:
-        raise RuntimeError(
+        raise ImportError(
             "onnx is not installed. Please install onnx to use this feature."
         )