minor unit test fix