Switching get overrides to use set_context