Fix imports after world breaking changes.