Fix: wrong name on class