Set correct context for find function