Minor update to a unit test