Fix context test