fix testclass