Fix import bug