diff --git a/src/test/java/com/github/streams/practice/numbers/NumbersProblemSolution.java b/src/test/java/com/github/streams/practice/numbers/NumbersProblemSolution.java
index c8addf7..0864643 100644
--- a/src/test/java/com/github/streams/practice/numbers/NumbersProblemSolution.java
+++ b/src/test/java/com/github/streams/practice/numbers/NumbersProblemSolution.java
@@ -40,7 +40,7 @@ public static <T> List<T> getUniqueNumberFromList(List<T> input) {
     return input.stream().distinct().toList();
   }
 
-  public static long sumOfNumbers(List<Integer> input) {
+  public static Integer sumOfNumbers(List<Integer> input) {
     return input.stream().reduce(0, Integer::sum);
   }
 
diff --git a/src/test/java/com/github/streams/practice/numbers/problems/A_SumOfNumbers.java b/src/test/java/com/github/streams/practice/numbers/problems/A_SumOfNumbers.java
index da25e38..ee9f12e 100644
--- a/src/test/java/com/github/streams/practice/numbers/problems/A_SumOfNumbers.java
+++ b/src/test/java/com/github/streams/practice/numbers/problems/A_SumOfNumbers.java
@@ -8,8 +8,8 @@
 
 /*
  * Given an array of {5,6,7,8,5,5,8,8,7)
- * Find the sum of the unique elements.
- * The output should be in this case is: 26.
+ * Find the sum of the all the elements.
+ * The output should be in this case is: 59.
  */
 class A_SumOfNumbers {
 
diff --git a/src/test/java/com/github/streams/practice/numbers/problems/I_SegregateEvenOddNumbers.java b/src/test/java/com/github/streams/practice/numbers/problems/I_SegregateEvenOddNumbers.java
index 10d8810..a05f9e1 100644
--- a/src/test/java/com/github/streams/practice/numbers/problems/I_SegregateEvenOddNumbers.java
+++ b/src/test/java/com/github/streams/practice/numbers/problems/I_SegregateEvenOddNumbers.java
@@ -16,10 +16,10 @@ class I_SegregateEvenOddNumbers {
   @Test
   @Disabled
   void testSegregationOfEvenOddNumbersTest() {
-    final var input = IntStream.range(1, 50).boxed();
-    final var mySolution = NumbersProblemSolution.segregateEvenOddNumbers(input);
+    final var input = IntStream.range(1, 50).boxed().toList();
+    final var mySolution = NumbersProblemSolution.segregateEvenOddNumbers(input.stream());
     final var yourSolution = new ArrayList<>();
-
+    
     Assertions.assertEquals(mySolution, yourSolution);
   }
 }