fixed button problem