Other unit test fix