Fix import error case