Fix for button alignment