fix issue #267

CamDavidsonPilon · Dec 28, 2016 · 8136156 · 8136156
1 parent 971782d
commit 8136156
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lifelines/plotting.py b/lifelines/plotting.py
@@ -374,7 +374,7 @@ def plot(ix=None, iloc=None, show_censors=False,
         if ci_show:
             if ci_force_lines:
                 dataframe_slicer(confidence_interval_).plot(linestyle="-", linewidth=1,
-                                                            color=[colour], legend=True,
+                                                            color=[colour], legend=ci_legend,
                                                             drawstyle=kwargs.get('drawstyle', 'default'),
                                                             ax=ax, alpha=0.6)
             else:
@@ -385,6 +385,7 @@ def plot(ix=None, iloc=None, show_censors=False,
                                    alpha=ci_alpha, color=colour,
                                    linewidth=1.0)
 
+
         if at_risk_counts:
             add_at_risk_counts(cls, ax=ax)