Fixing button to new styles