Heatmap focus jumps when clicking outside. Fix #4