Restrict diagrams version for now