diff --git a/src/data_structures/linked_lists/singly/MySinglyLinkedList.java b/src/data_structures/linked_lists/singly/MySinglyLinkedList.java index f34d00f..e3979cf 100644 --- a/src/data_structures/linked_lists/singly/MySinglyLinkedList.java +++ b/src/data_structures/linked_lists/singly/MySinglyLinkedList.java @@ -67,6 +67,10 @@ public void remove(int index) { MySinglyLinkedNode leader = traverseToIndex(index - 1); MySinglyLinkedNode nodeToRemove = leader.getNext(); leader.setNext(nodeToRemove.getNext()); + + if (index == length -1) { + tail = leader; + } length--; }