Button layout fix