Fix: name of class