MAINT: tweak import