small fixes in add addition flow