Unittest build fix