Fix and test with 2 agents