Fix button mouse handler