diff --git a/update_temporal_components_parallel.m b/update_temporal_components_parallel.m
index a54a37e..c3ff336 100644
--- a/update_temporal_components_parallel.m
+++ b/update_temporal_components_parallel.m
@@ -135,7 +135,7 @@
 end
 p = P.p;
 for iter = 1:ITER
-    [O,lo] = update_order(A);
+    [O,lo] = update_order(A(:,1:K));
     for jo = 1:length(O)
         Ytemp = YrA(:,O{jo}(:)) + (diag(nA(O{jo}))*Cin(O{jo},:))';
         Ctemp = zeros(length(O{jo}),T);
@@ -222,4 +222,4 @@
 f = C(K+1:end,:);
 C = C(1:K,:);
 Y_res(unsaturatedPix,:) = Y_res;
-Y_res(saturatedPix,:) = Ysat - Asat*C - bsat*f;
\ No newline at end of file
+Y_res(saturatedPix,:) = Ysat - Asat*C - bsat*f;