Change UnitTestExample to the recommended approach