diff --git a/image_analysis/IndexAndRefine.cpp b/image_analysis/IndexAndRefine.cpp
index f3cc65bf..ebe8f013 100644
--- a/image_analysis/IndexAndRefine.cpp
+++ b/image_analysis/IndexAndRefine.cpp
@@ -153,9 +153,9 @@ void IndexAndRefine::QuickPredictAndIntegrate(DataMessage &msg,
     BraggIntegrate2D(outcome.experiment, image, refl, msg.number);
 
     std::vector<Reflection> refl_ret;
-    refl_ret.reserve(prediction.GetReflections().size());
+    refl_ret.reserve(refl.size());
 
-    for (const auto& r : prediction.GetReflections()) {
+    for (const auto& r : refl) {
         if (r.observed)
             refl_ret.push_back(r);
     }