Small fixes in agi-test.go