Fix beat UI