flow: unit tests update