Make `Node::context` optional as well

This was an oversight in d09dfe0.

Co-Authored-By: Marshall <marshall@zed.dev>