minor unit test correction