Fixed bug with diagram labels