diff --git a/onnxmltools/convert/h2o/convert.py b/onnxmltools/convert/h2o/convert.py
index 3cc2f23f..24ba9001 100644
--- a/onnxmltools/convert/h2o/convert.py
+++ b/onnxmltools/convert/h2o/convert.py
@@ -10,7 +10,7 @@
 import h2o
 
 from onnxconverter_common.onnx_ex import get_maximum_opset_supported
-from ...proto import onnx
+import onnx
 from ..common._topology import convert_topology
 from ..common.data_types import FloatTensorType
 from ._parse import parse_h2o
diff --git a/tests/coreml/test_cml_AllNeuralNetworkConverters.py b/tests/coreml/test_cml_AllNeuralNetworkConverters.py
index c6272fb1..59511c66 100644
--- a/tests/coreml/test_cml_AllNeuralNetworkConverters.py
+++ b/tests/coreml/test_cml_AllNeuralNetworkConverters.py
@@ -8,12 +8,13 @@
         setattr(sklearn.preprocessing, 'Imputer', Imputer)
 except ImportError:
     from sklearn.preprocessing import Imputer
+import onnx
 from coremltools.models.neural_network import NeuralNetworkBuilder
 from coremltools.models import datatypes
 from coremltools.proto.FeatureTypes_pb2 import ImageFeatureType
 from distutils.version import StrictVersion
 from onnxmltools import convert_coreml
-from onnxmltools.proto import onnx
+
 
 class TestNeuralNetworkLayerConverter(unittest.TestCase):