diagram fixes