finally fixed button I think