fixes diagram