Fixed issue with selected nodes not getting checked.