1 files changed, 3 insertions, 0 deletions
diff --git a/internal/net/server.go b/internal/net/server.go
index 8114df2..851a4b5 100644
--- a/internal/net/server.go
+++ b/internal/net/server.go
@@ -45,6 +45,7 @@ const (
 	StateBank
 	StateHacking
 	StateDangerConfirm
+	StateUndigConfirm
 )
 
 type Session struct {
@@ -62,6 +63,8 @@ type Session struct {
 	Disconnecting     bool
 	DisconnectTicks   int
 	Shop              *behavior.ShopConfig
+	PendingUndigDir   string
+	PendingUndigRoom  int
 	promptVisible     bool
 	writeMu           sync.Mutex
 }