struct arm_smmu_device *smmu = platform_get_drvdata(pdev);
 
        arm_smmu_device_disable(smmu);
+
        return 0;
 }
 
+static void arm_smmu_device_shutdown(struct platform_device *pdev)
+{
+       arm_smmu_device_remove(pdev);
+}
+
 static const struct of_device_id arm_smmu_of_match[] = {
        { .compatible = "arm,smmu-v3", },
        { },
        },
        .probe  = arm_smmu_device_probe,
        .remove = arm_smmu_device_remove,
+       .shutdown = arm_smmu_device_shutdown,
 };
 module_platform_driver(arm_smmu_driver);
 
 
        return 0;
 }
 
+static void arm_smmu_device_shutdown(struct platform_device *pdev)
+{
+       arm_smmu_device_remove(pdev);
+}
+
 static struct platform_driver arm_smmu_driver = {
        .driver = {
                .name           = "arm-smmu",
        },
        .probe  = arm_smmu_device_probe,
        .remove = arm_smmu_device_remove,
+       .shutdown = arm_smmu_device_shutdown,
 };
 module_platform_driver(arm_smmu_driver);