Fix: selection after removing node