Minor import tweaks