1 files changed, 3 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c b/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
index 7c4ec108..5b43eb4b 100644
--- a/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
@@ -620,10 +620,11 @@ long gk20a_ctrl_dev_ioctl(struct file *filp, unsigned int cmd, unsigned long arg
 
                 if (!err) {
                         err = gk20a_busy(dev);
-                        if (!err)
+                        if (!err) {
                                 err = gk20a_gr_zbc_set_table(g, &g->gr,
                                                              zbc_val);
-                        gk20a_idle(dev);
+                                gk20a_idle(dev);
+                        }
                 }
 
                 if (zbc_val)