diff --git a/pkg/models/state.go b/pkg/models/state.go
index 338efd9..a1cd254 100644
--- a/pkg/models/state.go
+++ b/pkg/models/state.go
@@ -15,13 +15,10 @@ type SessionType string
 type RemoteAddr string
 
 type Agent struct {
-	Guid       AgentGuid
-	RemoteAddr RemoteAddr
-	PublicId   RendezVousId
-	StartTime  time.Time
-
-	// TODO add remote address.
-
+	Guid            AgentGuid
+	RemoteAddr      RemoteAddr
+	PublicId        RendezVousId
+	StartTime       time.Time
 	EnvironmentInfo comms.EnvironmentInfo
 	expiryTime      int64
 }