Changed the check for the return flow value.