Fix: Wrong class name