Ensure token map is rebuilt when nodes are removed