Fix TestBase