fix return from update