Fixes issue #17 - bad diagram