Allow passing in context argument.