diff --git a/source/poet.cpp b/source/poet.cpp
index 9efb219..ee06680 100644
--- a/source/poet.cpp
+++ b/source/poet.cpp
@@ -91,7 +91,7 @@ int main()
 	arma::mat next;
 	std::string curr_str;
 	std::string next_str;
-	for (int i=0; i < t_vc.n_rows-50; i++)
+	for (int i=0; i < stack->getSeqLen(); i++)
 	{
 		curr = stack->to_curr(t_vc.row(i));
 		curr_str.append(1,RnnTextHelper::idx2ch(curr.index_max()));
@@ -102,7 +102,7 @@ int main()
 	cout << "Curr: " << curr_str << std::endl;
 	cout << "Next: " << next_str << std::endl;
 	
-	for (int i=0; i < 120; i++)
+	for (int i=stack->getSeqLen(); i < t_vc.n_rows; i++)
 	{
 		curr = next;
 		curr_str.append(1,RnnTextHelper::idx2ch(curr.index_max()));