fix bug: wrong class name