Fix diagram