Fix the main() node.

This makes the graph have a more predictable shape to it.