Fix UI "draw beat" feature