Fix : pass correct class name