plot_completion_diagram: Make plot_walk_in_completion_diagram work again