Fix for button