Fix unitest compile error