Make sure the context gets set