Update for new layout of IMP.pmi.io module.