Fix BUtton going away or not working