diff --git a/__main__.py b/__main__.py
index ce5ac570eac7d396ea38483d51b5780eed0b0bb2..dd894ceb9a920d3514fcf75166c168440fa3d316 100644
--- a/__main__.py
+++ b/__main__.py
@@ -90,11 +90,12 @@ def split_leaf(node, key):
         median_value = node.keys[median_index - 1]
         redistribute_keys(node, right_node, median_index - 1, median_index - 1)
         array_insert_sorted(node.keys, key)
+    elif virtual_insertion_index > median_index:
+        median_value = node.keys[median_index]
+        redistribute_keys(node, right_node, median_index, median_index)
+        array_insert_sorted(right_node.keys, key)
     else:
-        if virtual_insertion_index > median_index:
-            median_value = node.keys[median_index]
-        else:
-            median_value = key
+        median_value = key
         redistribute_keys(node, right_node, median_index, median_index)
         array_insert_sorted(right_node.keys, key)