Switch to using a .get for checking context.