Fixed a bug in class diagrams