diff --git a/src/abstract_constraint.jl b/src/abstract_constraint.jl
index 4d9891fb..52c1eff6 100644
--- a/src/abstract_constraint.jl
+++ b/src/abstract_constraint.jl
@@ -190,7 +190,7 @@ All constraints are categorized into the following type tree:
                         ↙                ↘
            StageConstraint               CoupledConstraint
             ↙        ↘                       ↙           ↘
-StageConstraint ControlConstraint CoupledStateConstraint CoupledControlConstraint
+StateConstraint ControlConstraint CoupledStateConstraint CoupledControlConstraint
 ```
 
 The state and control dimensions (where applicable) can be queried using
@@ -570,4 +570,4 @@ function gen_views(
     else
         view(∇c, :, 1:n), view(∇c, :, n .+ (1:m))
     end
-end
\ No newline at end of file
+end