diff --git a/drivers/gpu/nvgpu/gp106/hal_gp106.c b/drivers/gpu/nvgpu/gp106/hal_gp106.c
index 9afcdb696..cdac76b70 100644
--- a/drivers/gpu/nvgpu/gp106/hal_gp106.c
+++ b/drivers/gpu/nvgpu/gp106/hal_gp106.c
@@ -98,8 +98,7 @@ static struct gpu_ops gp106_ops = {
 		}
 };
 
-static int gp106_get_litter_value(struct gk20a *g,
-		enum nvgpu_litter_value value)
+static int gp106_get_litter_value(struct gk20a *g, int value)
 {
 	int ret = -EINVAL;
 
diff --git a/drivers/gpu/nvgpu/gp10b/hal_gp10b.c b/drivers/gpu/nvgpu/gp10b/hal_gp10b.c
index f0137a709..e9385db0b 100644
--- a/drivers/gpu/nvgpu/gp10b/hal_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/hal_gp10b.c
@@ -105,8 +105,7 @@ static struct gpu_ops gp10b_ops = {
 	}
 };
 
-static int gp10b_get_litter_value(struct gk20a *g,
-		enum nvgpu_litter_value value)
+static int gp10b_get_litter_value(struct gk20a *g, int value)
 {
 	int ret = EINVAL;
 	switch (value) {