Adjust this to grab the context as well