diff --git a/algorithms-modules/algorithms-miscellaneous-5/src/main/java/com/baeldung/algorithms/integerstreammedian/MedianOfIntegerStream.java b/algorithms-modules/algorithms-miscellaneous-5/src/main/java/com/baeldung/algorithms/integerstreammedian/MedianOfIntegerStream.java
index 9d3b699956..d3fde8a827 100644
--- a/algorithms-modules/algorithms-miscellaneous-5/src/main/java/com/baeldung/algorithms/integerstreammedian/MedianOfIntegerStream.java
+++ b/algorithms-modules/algorithms-miscellaneous-5/src/main/java/com/baeldung/algorithms/integerstreammedian/MedianOfIntegerStream.java
@@ -29,13 +29,13 @@ public class MedianOfIntegerStream {
     }
 
     double getMedian() {
-        int median;
+        double median;
         if (minHeap.size() < maxHeap.size()) {
             median = maxHeap.peek();
         } else if (minHeap.size() > maxHeap.size()) {
             median = minHeap.peek();
         } else {
-            median = (minHeap.peek() + maxHeap.peek()) / 2;
+            median = (minHeap.peek() + maxHeap.peek()) / 2.0;
         }
         return median;
     }
diff --git a/algorithms-modules/algorithms-miscellaneous-5/src/main/java/com/baeldung/algorithms/integerstreammedian/MedianOfIntegerStream2.java b/algorithms-modules/algorithms-miscellaneous-5/src/main/java/com/baeldung/algorithms/integerstreammedian/MedianOfIntegerStream2.java
index 8aa7e6f6af..a28709c2d6 100644
--- a/algorithms-modules/algorithms-miscellaneous-5/src/main/java/com/baeldung/algorithms/integerstreammedian/MedianOfIntegerStream2.java
+++ b/algorithms-modules/algorithms-miscellaneous-5/src/main/java/com/baeldung/algorithms/integerstreammedian/MedianOfIntegerStream2.java
@@ -25,11 +25,11 @@ public class MedianOfIntegerStream2 {
     }
 
     double getMedian() {
-        int median;
+        double median;
         if (minHeap.size() > maxHeap.size()) {
             median = minHeap.peek();
         } else {
-            median = (minHeap.peek() + maxHeap.peek()) / 2;
+            median = (minHeap.peek() + maxHeap.peek()) / 2.0;
         }
         return median;
     }