Fix bug in PE module