diff --git a/azure-iot-device/azure/iot/device/iothub/abstract_clients.py b/azure-iot-device/azure/iot/device/iothub/abstract_clients.py
index 24a819d07..c09349373 100644
--- a/azure-iot-device/azure/iot/device/iothub/abstract_clients.py
+++ b/azure-iot-device/azure/iot/device/iothub/abstract_clients.py
@@ -17,7 +17,8 @@
 from azure.iot.device.common.auth import connection_string as cs
 from azure.iot.device.common.auth import sastoken as st
 from azure.iot.device.iothub import client_event
-from azure.iot.device import exceptions, Message, MethodRequest, X509
+from azure.iot.device.iothub.models import Message, MethodRequest
+from azure.iot.device import exceptions, X509
 from azure.iot.device.common import auth, handle_exceptions
 from . import edge_hsm
 from .pipeline import MQTTPipeline, HTTPPipeline