diff --git a/Sources/ChangeableMacros/ChangeableMacro.swift b/Sources/ChangeableMacros/ChangeableMacro.swift
index 5f84650..556d20a 100644
--- a/Sources/ChangeableMacros/ChangeableMacro.swift
+++ b/Sources/ChangeableMacros/ChangeableMacro.swift
@@ -66,7 +66,7 @@ public struct ChangeableFunctionMacro: MemberMacro {
     .joined(separator: ",\n    ")
 
     let applyDeclaration: DeclSyntax = """
-    public func apply(action: SetValue<State, some Any>) -> Self {
+    public func apply(action: SetValue<Self, some Any>) -> Self {
       let path = action.path
       let value = action.value
 
diff --git a/Tests/ChangeableTests/Resources/testChangeable/Output.swift.test b/Tests/ChangeableTests/Resources/testChangeable/Output.swift.test
index 01c1926..fdffdd3 100644
--- a/Tests/ChangeableTests/Resources/testChangeable/Output.swift.test
+++ b/Tests/ChangeableTests/Resources/testChangeable/Output.swift.test
@@ -15,7 +15,7 @@ b != nil
       )
     }
 
-    public func apply(action: SetValue<State, some Any>) -> Self {
+    public func apply(action: SetValue<Self, some Any>) -> Self {
       let path = action.path
       let value = action.value
 
diff --git a/Tests/ChangeableTests/Resources/testChangeableWithComment/Output.swift.test b/Tests/ChangeableTests/Resources/testChangeableWithComment/Output.swift.test
index 7584330..e290978 100644
--- a/Tests/ChangeableTests/Resources/testChangeableWithComment/Output.swift.test
+++ b/Tests/ChangeableTests/Resources/testChangeableWithComment/Output.swift.test
@@ -18,7 +18,7 @@ b != nil
       )
     }
 
-    public func apply(action: SetValue<State, some Any>) -> Self {
+    public func apply(action: SetValue<Self, some Any>) -> Self {
       let path = action.path
       let value = action.value