Small fix in class name