Fix diagram spacing