Fix parent class name of tests