Fix Test class