Fixing the janky buttons