diff --git a/hash_objects/hash_set_bst.py b/hash_objects/hash_set_bst.py
index 9f3b351..08cb8b4 100644
--- a/hash_objects/hash_set_bst.py
+++ b/hash_objects/hash_set_bst.py
@@ -82,7 +82,7 @@ class BSTree():
             root = root.right
         return root.value
 
-    def delete(self, root, key) -> TreeNode:
+    def delete(self, root, key) -> Node:
         if not root:
             return None