Fix the test base class