diff --git a/drivers/iommu/rockchip-iommu.c b/drivers/iommu/rockchip-iommu.c
index ad914dd17443..c4c2c54a6127 100644
--- a/drivers/iommu/rockchip-iommu.c
+++ b/drivers/iommu/rockchip-iommu.c
@@ -1216,7 +1216,7 @@ static int rk_iommu_probe(struct platform_device *pdev)
 
 static int rk_iommu_remove(struct platform_device *pdev)
 {
-	pm_runtime_put(&pdev->dev);
+	pm_runtime_disable(&pdev->dev);
 
 	return 0;
 }