minor changes to unit test