Replace more complex usages of within_context with $$.
Turn within_context() into an expression.