diff --git a/drivers/clk/rockchip/clk_rk3399.c b/drivers/clk/rockchip/clk_rk3399.c
index ea0ce2aab1..8efb892606 100644
--- a/drivers/clk/rockchip/clk_rk3399.c
+++ b/drivers/clk/rockchip/clk_rk3399.c
@@ -786,7 +786,9 @@ void *rockchip_get_cru(void)
 	fdt_addr_t *addr;
 	int ret;
 
-	ret = uclass_get_device_by_name(UCLASS_CLK, "clk_rk3399", &dev);
+	ret = uclass_get_device_by_driver(UCLASS_CLK,
+			DM_GET_DRIVER(clk_rk3399), &dev);
+
 	if (ret)
 		return ERR_PTR(ret);