fix buttons bug