diff --git a/api/edge/edge.go b/api/edge/edge.go
index adafecec5..38cb469c7 100644
--- a/api/edge/edge.go
+++ b/api/edge/edge.go
@@ -25,6 +25,8 @@ type (
 		Namespace string
 		// Version of the stack file
 		Version int
+		// RollbackTo specifies the stack file version to rollback to (only support to rollback to the last version currently)
+		RollbackTo *int
 
 		// RegistryCredentials holds the credentials for a Docker registry.
 		// Used only for EE
diff --git a/api/portainer.go b/api/portainer.go
index 1c672f74d..2fb1e8957 100644
--- a/api/portainer.go
+++ b/api/portainer.go
@@ -364,6 +364,8 @@ type (
 		Time  int64
 		Type  EdgeStackStatusType
 		Error string
+		// EE only feature
+		RollbackTo *int
 	}
 
 	//EdgeStackStatusType represents an edge stack status type