Fixed references to new diagram library