return ret;
 }
 
-static int vpu_core_remove(struct platform_device *pdev)
+static void vpu_core_remove(struct platform_device *pdev)
 {
        struct device *dev = &pdev->dev;
        struct vpu_core *core = platform_get_drvdata(pdev);
        memunmap(core->rpc.virt);
        mutex_destroy(&core->lock);
        mutex_destroy(&core->cmd_lock);
-
-       return 0;
 }
 
 static int __maybe_unused vpu_core_runtime_resume(struct device *dev)
 
 static struct platform_driver amphion_vpu_core_driver = {
        .probe = vpu_core_probe,
-       .remove = vpu_core_remove,
+       .remove_new = vpu_core_remove,
        .driver = {
                .name = "amphion-vpu-core",
                .of_match_table = vpu_core_dt_match,