Fix: wrong class name