diff --git a/azure-iot-device/azure/iot/device/iothub/models/methods.py b/azure-iot-device/azure/iot/device/iothub/models/methods.py
index b88b29a4d..53fea1475 100644
--- a/azure-iot-device/azure/iot/device/iothub/models/methods.py
+++ b/azure-iot-device/azure/iot/device/iothub/models/methods.py
@@ -5,7 +5,8 @@
 # --------------------------------------------------------------------------
 """This module contains classes related to direct method invocations.
 """
-from typing import Optional, Self
+from typing import Optional
+from typing_extensions import Self
 from azure.iot.device.custom_typing import JSONSerializable