Update go flow. Fix bug.