diff --git a/snippets/median.md b/snippets/median.md
index 63119ad66..6de64941b 100644
--- a/snippets/median.md
+++ b/snippets/median.md
@@ -14,10 +14,10 @@ def median(list):
   list_length = len(list)
   if list_length % 2 == 0:
     return (list[int(list_length / 2) - 1] + list[int(list_length / 2)]) / 2
-  return list[int(list_length / 2)]
+  return float(list[int(list_length / 2)])
 ```
 
 ```py
-median([1,2,3]) # 2
+median([1,2,3]) # 2.0
 median([1,2,3,4]) # 2.5
 ```