Allow key functions to self.

[kalekundert]

Cast functions can do this by annotating their argument being a Context object. I could do the same thing here...