Fix tests to check the correct node.

Trivial fix to these tests to check the important node is instantiated.