MNT: Updated import to match api change in metadatastore