diff --git a/src/hackerrank/interview_preparation_kit/search/swap_nodes_algo.py b/src/hackerrank/interview_preparation_kit/search/swap_nodes_algo.py
index 2813f922..e40c1d1e 100644
--- a/src/hackerrank/interview_preparation_kit/search/swap_nodes_algo.py
+++ b/src/hackerrank/interview_preparation_kit/search/swap_nodes_algo.py
@@ -79,7 +79,7 @@ def buildTree(indexes: List[List[int]]) -> Node:
             __INITIAL_LEVEL__,
             callbackCollectNodes)
 
-        last_level: int = sorted(node_collector)[-1]
+        last_level: int = max(node_collector)
 
         for i in range(0, min(len(indexes_copy), len(node_collector[last_level]))):
             current_node: Node = node_collector[last_level][i]