Actual fix for button