diff --git a/experiments/matrix-completion-accuracy.py b/experiments/matrix-completion-accuracy.py
index 1e3240a007e408cba8d2313a53438a1068f994e8..54609d80cb0b8653af68f8e89e0572c753934a77 100644
--- a/experiments/matrix-completion-accuracy.py
+++ b/experiments/matrix-completion-accuracy.py
@@ -29,7 +29,7 @@ import numpy as np
 import pandas as pd
 
 from dataset_paths import PETERS2009_CSV_PATH
-from mhcflurry.score_set import ScoreSet
+from score_set import ScoreSet
 from mhcflurry.dataset import Dataset
 from matrix_completion_helpers import (
     evaluate_predictions,