Fixed missing $go, and added test for it.