Fix pshb migration