Base class for mock tests
Minor fix