diff --git a/Sources/Models/Value.swift b/Sources/Models/Value.swift
index cdfc4c6..f9b4d3a 100644
--- a/Sources/Models/Value.swift
+++ b/Sources/Models/Value.swift
@@ -67,11 +67,6 @@ public enum FastisModeRange {
     case range
 }
 
-public enum DismissAction {
-    case done(FastisValue?)
-    case cancel
-}
-
 extension Date: FastisValue {
 
     /// Mode of value for ``FastisController``. Always `.single`
diff --git a/Sources/Views/Controller.swift b/Sources/Views/Controller.swift
index 3e907a3..e9823ea 100644
--- a/Sources/Views/Controller.swift
+++ b/Sources/Views/Controller.swift
@@ -769,3 +769,18 @@ public extension FastisConfig {
 
     }
 }
+
+public extension FastisController {
+
+    /**
+     Parameter to return in the dismissHandler
+
+     `.done(Value?)` - If a date is selected.
+     `.cancel` - if controller closed without date selection
+     */
+
+    enum DismissAction {
+        case done(Value?)
+        case cancel
+    }
+}