diff --git a/openemc-driver/openemc_supply.c b/openemc-driver/openemc_supply.c
index 9745f5c..077022c 100644
--- a/openemc-driver/openemc_supply.c
+++ b/openemc-driver/openemc_supply.c
@@ -182,7 +182,7 @@ static ssize_t openemc_supply_connect_data_show(struct device *dev,
 						struct device_attribute *attr,
 						char *buf)
 {
-	struct openemc_supply *sup = dev_get_drvdata(dev->parent);
+	struct openemc_supply *sup = dev_get_drvdata(dev);
 	int ret;
 	u8 connect;
 
@@ -199,7 +199,7 @@ static ssize_t openemc_supply_connect_data_store(struct device *dev,
 						 struct device_attribute *attr,
 						 const char *buf, size_t count)
 {
-	struct openemc_supply *sup = dev_get_drvdata(dev->parent);
+	struct openemc_supply *sup = dev_get_drvdata(dev);
 	u8 connect;
 	int ret;