fixed failing test, typo
class not define
fixed failing tests