Unit tests fix for PE