diff --git a/source/poet.cpp b/source/poet.cpp
index 76a0c4e..a06a3cd 100644
--- a/source/poet.cpp
+++ b/source/poet.cpp
@@ -178,7 +178,7 @@ int main()
 
 	h = layer->toHiddenProbs(v.row(0));
 	printf("\nStimulus: Next char and 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);