diff --git a/drivers/thermal/thermal_netlink.c b/drivers/thermal/thermal_netlink.c
index 0bfe30233afe..c82dba5ee231 100644
--- a/drivers/thermal/thermal_netlink.c
+++ b/drivers/thermal/thermal_netlink.c
@@ -232,7 +232,7 @@ static int thermal_genl_send_event(enum thermal_genl_event event,
 	void *hdr;
 	int enable_thermal_genl = 1;
 
-	trace_android_vh_enable_thermal_genl_check(event, &enable_thermal_genl);
+	trace_android_vh_enable_thermal_genl_check(event, p->tz_id, &enable_thermal_genl);
 	if (!enable_thermal_genl)
 		return 0;
 
diff --git a/include/trace/hooks/thermal.h b/include/trace/hooks/thermal.h
index 0e47e0af9284..ae9b7ee652fb 100644
--- a/include/trace/hooks/thermal.h
+++ b/include/trace/hooks/thermal.h
@@ -11,8 +11,8 @@
 #include <trace/hooks/vendor_hooks.h>
 
 DECLARE_HOOK(android_vh_enable_thermal_genl_check,
-	TP_PROTO(int event, int *enable_thermal_genl),
-	TP_ARGS(event, enable_thermal_genl));
+	TP_PROTO(int event, int tz_id, int *enable_thermal_genl),
+	TP_ARGS(event, tz_id, enable_thermal_genl));
 
 #endif /* _TRACE_HOOK_THERMAL_H */
 /* This part must be outside protection */