change test base class back to avoid error