Updated the tests context creation.