improve pce unit tests