diff --git a/matlab/particle/traditional_resampling.m b/matlab/particle/traditional_resampling.m
index 49110ecc43445a796dbe74834b6147a7eaab1bd3..3e5d38e909a7f93e668cc211172587c4ebc4de30 100644
--- a/matlab/particle/traditional_resampling.m
+++ b/matlab/particle/traditional_resampling.m
@@ -1,4 +1,4 @@
-function indx = traditional_resampling(weights,noise)
+function return_resample = traditional_resampling(particles,weights,noise)
 % Resamples particles.
 
 %@info:
@@ -96,6 +96,12 @@ else
     indx = indx+1;
 end
 
+if particles==0
+    return_resample = indx ;
+else
+    return_resample = particles(indx,:) ;
+end
+
 %@test:1
 %$ % Define the weights
 %$ weights = randn(2000,1).^2;