- fixing button update state