corrected test class