Fix import warning