diff --git a/corelib/src/BayesFilter.cpp b/corelib/src/BayesFilter.cpp
index aea1b7df..28c80b0c 100644
--- a/corelib/src/BayesFilter.cpp
+++ b/corelib/src/BayesFilter.cpp
@@ -273,7 +273,7 @@ cv::Mat BayesFilter::generatePrediction(const Memory * memory, const std::vector
 					{
 						if(iter->second == 0)
 						{
-							idsLoopMargin.push_back(iter->second);
+							idsLoopMargin.push_back(iter->first);
 						}
 						++iter;
 					}