diff --git a/ddmc/figures/figureM5.py b/ddmc/figures/figureM5.py
index 7b162022..63f36f29 100644
--- a/ddmc/figures/figureM5.py
+++ b/ddmc/figures/figureM5.py
@@ -44,7 +44,6 @@ def makeFigure():
         hue_loadings_title="p < 0.01",
     )
 
-    print(is_tumor)
     plot_p_signal_across_clusters_and_binary_feature(
         is_tumor, centers, "is_tumor", axes[3]
     )
diff --git a/ddmc/logistic_regression.py b/ddmc/logistic_regression.py
index c7084236..7af733e8 100644
--- a/ddmc/logistic_regression.py
+++ b/ddmc/logistic_regression.py
@@ -71,8 +71,8 @@ def plot_roc(
     mean_fpr = np.linspace(0, 1, 100)
 
     for _, (train, test) in enumerate(cv.split(X, y)):
-        classifier.fit(X[train], y[train])
-        viz = RocCurveDisplay.from_estimator(classifier, X[test], y[test])
+        classifier.fit(X.iloc[train], y.iloc[train])
+        viz = RocCurveDisplay.from_estimator(classifier, X.iloc[test], y.iloc[test])
         plt.close()
         interp_tpr = np.interp(mean_fpr, viz.fpr, viz.tpr)
         interp_tpr[0] = 0.0