Fix comile error for tests