Fix VPE tests