Diagram floating issue fixed