Fix import bug (again...)