Removed unused import from diagram