Removed bug where nodes weren't being marked.