fix test for the context