diff --git a/matlab/DiffuseKalmanSmootherH3_Z.m b/matlab/DiffuseKalmanSmootherH3_Z.m
index 1fbc193902463dd6e3e7b5e9a5976eca65b9effe..5dc6109a051aa27a893851167d5473b6f74337cd 100644
--- a/matlab/DiffuseKalmanSmootherH3_Z.m
+++ b/matlab/DiffuseKalmanSmootherH3_Z.m
@@ -254,7 +254,7 @@ while t>d+1
       ri = Z(i,:)'/Fi(i,t)*v(i,t)+Li(:,:,i,t)'*ri;
     end
   end
-  r(:,t) = ri(:,t);
+  r(:,t) = ri;
   alphahat(:,t) = a1(:,t) + P1(:,:,t)*r(:,t);
   etahat(:,t) = QRt*r(:,t);
   ri = T'*ri;