diff --git a/drivers/gpu/nvgpu/common/gr/gr_ecc.c b/drivers/gpu/nvgpu/common/gr/gr_ecc.c
index bdad8a6c2..de47b9e00 100644
--- a/drivers/gpu/nvgpu/common/gr/gr_ecc.c
+++ b/drivers/gpu/nvgpu/common/gr/gr_ecc.c
@@ -91,7 +91,7 @@ int nvgpu_ecc_counter_init_per_tpc(struct gk20a *g,
 
 fail:
 	if (err != 0) {
-#ifdef CONFIG_NVGPU_NON_FUSA
+#ifdef CONFIG_NVGPU_DGPU
 		while (gpc-- != 0u) {
 			nvgpu_kfree(g, stats[gpc]);
 		}