Fix context tests