fix unit test context