diff --git a/python/Chapter 2/classes/LinkedList.py b/python/Chapter 2/classes/LinkedList.py index fbde35d7..8ad0ba71 100644 --- a/python/Chapter 2/classes/LinkedList.py +++ b/python/Chapter 2/classes/LinkedList.py @@ -42,6 +42,8 @@ def removeNode(self, node_value): else: while(current.next != None): if current.next.value == node_value: + if current.next == self.tail: + self.tail = current current.next = current.next.next break else: