Complete the unit tests on the agent