From 33d3c9124bc2d9129551f6f03d2db75e2ce6f930 Mon Sep 17 00:00:00 2001
From: Jake Taylor <jake@elixxir.io>
Date: Fri, 28 Jan 2022 12:43:05 -0600
Subject: [PATCH] id space fix

---
 network/follow.go  | 5 +++++
 network/manager.go | 1 +
 2 files changed, 6 insertions(+)

diff --git a/network/follow.go b/network/follow.go
index 89df4f805..5f3491587 100644
--- a/network/follow.go
+++ b/network/follow.go
@@ -223,6 +223,11 @@ func (m *manager) follow(report interfaces.ClientErrorReport, rng csprng.Source,
 		}
 	}
 
+	// Update the address space size
+	if len(m.Instance.GetPartialNdf().Get().AddressSpace) != 0 {
+		m.addrSpace.Update(m.Instance.GetPartialNdf().Get().AddressSpace[0].Size)
+	}
+
 	// NOTE: this updates rounds and updates the tracking of the health of the network
 	if pollResp.Updates != nil {
 		// TODO: ClientErr needs to know the source of the error and it doesn't yet
diff --git a/network/manager.go b/network/manager.go
index c968102e2..bba264c6b 100644
--- a/network/manager.go
+++ b/network/manager.go
@@ -105,6 +105,7 @@ func NewManager(session *storage.Session, switchboard *switchboard.Switchboard,
 		events:        events,
 		earliestRound: &earliest,
 	}
+	m.addrSpace.Update(18)
 
 	if params.VerboseRoundTracking {
 		m.verboseRounds = NewRoundTracker()
-- 
GitLab