minor fix class diagram