fix(agent): Report UnitState only when appropriate

Fix #446