Fix unitest