diff --git a/custom_components/saj_modbus/hub.py b/custom_components/saj_modbus/hub.py
index c6a7517..1bac6e2 100644
--- a/custom_components/saj_modbus/hub.py
+++ b/custom_components/saj_modbus/hub.py
@@ -1,6 +1,6 @@
 """SAJ Modbus Hub"""
-from pymodbus.register_read_message import ReadHoldingRegistersResponse
-from pymodbus.register_write_message import ModbusResponse
+from pymodbus.pdu.register_read_message import ReadHoldingRegistersResponse
+from pymodbus.pdu.register_write_message import ModbusResponse
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
 from voluptuous.validators import Number
 import logging