Minor import fix.