Correct passing context between tests