Improve context lost test