minor change in unit test