diff --git a/include/linux/host1x.h b/include/linux/host1x.h
index b5358f855d9e..476da0e06bb2 100644
--- a/include/linux/host1x.h
+++ b/include/linux/host1x.h
@@ -249,7 +249,7 @@ struct host1x_job {
 	u8 *gather_copy_mapped;
 
 	/* Check if register is marked as an address reg */
-	int (*is_addr_reg)(struct device *dev, u32 reg, u32 class);
+	int (*is_addr_reg)(struct device *dev, u32 class, u32 reg);
 
 	/* Check if class belongs to the unit */
 	int (*is_valid_class)(u32 class);