Improve test for context preservation

currently the test does not go past the first check