Fix type in class name