unit test fixes