Fix to compilation error in test