fix the test class

not running