Trying to fix issue with button