add class
fix testcase