Fix tests to check the correct node.
Merge pull request #35 from hindessm/fix-tests

Fix tests to check the correct node.