Fix for Diagram type change that I missed