1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/module.c b/drivers/gpu/nvgpu/common/linux/module.c
index cbfe6ad7..bfbe7a58 100644
--- a/drivers/gpu/nvgpu/common/linux/module.c
+++ b/drivers/gpu/nvgpu/common/linux/module.c
@@ -659,7 +659,7 @@ static void gk20a_pm_shutdown(struct platform_device *pdev)
         /* Prevent more requests by disabling Runtime PM */
         __pm_runtime_disable(&pdev->dev, false);
 
-        err = gk20a_wait_for_idle(&pdev->dev);
+        err = gk20a_wait_for_idle(g);
         if (err) {
                 nvgpu_err(g, "failed to idle GPU, err=%d", err);
                 goto finish;
@@ -844,7 +844,7 @@ void gk20a_driver_start_unload(struct gk20a *g)
         if (g->is_virtual)
                 return;
 
-        gk20a_wait_for_idle(dev_from_gk20a(g));
+        gk20a_wait_for_idle(g);
 
         nvgpu_wait_for_deferred_interrupts(g);
         gk20a_channel_cancel_pending_sema_waits(g);