1 files changed, 2 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/fecs_trace_vgpu.c b/drivers/gpu/nvgpu/vgpu/fecs_trace_vgpu.c
index 02e0baad..bd1a7451 100644
--- a/drivers/gpu/nvgpu/vgpu/fecs_trace_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/fecs_trace_vgpu.c
@@ -58,6 +58,7 @@ static int vgpu_fecs_trace_init(struct gk20a *g)
         if (IS_ERR(vcst->cookie)) {
                 dev_info(dev_from_gk20a(g),
                         "mempool  %u reserve failed\n", mempool);
+                vcst->cookie = NULL;
                 err = -EINVAL;
                 goto fail;
         }
@@ -85,7 +86,7 @@ static int vgpu_fecs_trace_init(struct gk20a *g)
         return 0;
 fail:
         iounmap(vcst->buf);
-        if (!IS_ERR(vcst->cookie))
+        if (vcst->cookie)
                 tegra_hv_mempool_unreserve(vcst->cookie);
         kfree(vcst);
         return err;