Fix regression: allow args to be lambdas again:

* Also allow this in control args too.