Fix button action