Fix selection test logic.

Selection test logic is updated to match new returns from selection
nodes.