fix most agent class tests