fix issue with button