Fix for branch updates in PE