Fix class name, whoops