Function arrow choice ValueError now have correct detail message.

1 files changed, 2 insertions, 2 deletions

diff --git a/src/pypeline/core/arrows/function_arrow_choice.py b/src/pypeline/core/arrows/function_arrow_choice.py
index c5ef6a3..a90326b 100644
--- a/src/pypeline/core/arrows/function_arrow_choice.py
+++ b/src/pypeline/core/arrows/function_arrow_choice.py
@@ -74,7 +74,7 @@ class FunctionArrowChoice(ArrowChoice, FunctionArrow):
     #                 +---------------+                   +---------------+
     def __add__(self, other):
         if not isinstance(other, FunctionArrowChoice):
-            raise ValueError("Must be a FunctionArrow")
+            raise ValueError("Must be a FunctionArrowChoice")
 
         return self.left() >> other.right()
 
@@ -88,7 +88,7 @@ class FunctionArrowChoice(ArrowChoice, FunctionArrow):
     #                 +---------------+                   +---------------+                   +-------------------+
     def __or__(self, other):
         if not isinstance(other, FunctionArrowChoice):
-            raise ValueError("Must be a FunctionArrow")
+            raise ValueError("Must be a FunctionArrowChoice")
 
         def merge(either):
             if not isinstance(either, Either):