diff --git a/source/poet.cpp b/source/poet.cpp
index a06a3cd..f827121 100644
--- a/source/poet.cpp
+++ b/source/poet.cpp
@@ -193,7 +193,7 @@ int main()
 
 	h = layer->toHiddenProbs(v.row(0));
 	printf("\nStimulus: Current h\n");
-	for (int i=1; i < numTraining; i++)
+	for (int i=1; i < numTraining+1; i++)
 	{
 		layer->gibbs_hv(h, r);
 		arma::mat curr = to_curr(r);