Fix event declaration in heatmap too [#4]

Signed-off-by: Kirill Enykeev <enykeev@gmail.com>