Fix up bad import