Fixing the test context setup.