Fix PM assignment after migration