diff --git a/internal/state/config/config.go b/internal/state/config/config.go
index fe557fd0..9677d0ef 100644
--- a/internal/state/config/config.go
+++ b/internal/state/config/config.go
@@ -141,8 +141,9 @@ func (cfg *config) marshal() ([]byte, error) {
 	}
 	for _, context := range cfg.contexts {
 		raw.Contexts = append(raw.Contexts, rawConfigContext{
-			Name:  context.Name,
-			Token: context.Token,
+			Name:    context.Name,
+			Token:   context.Token,
+			SSHKeys: context.SSHKeys,
 		})
 	}
 	return toml.Marshal(raw)