diff --git a/source/poet.cpp b/source/poet.cpp
index bbb69c5..881590e 100644
--- a/source/poet.cpp
+++ b/source/poet.cpp
@@ -81,8 +81,8 @@ int main()
 	
 	Stack stack(".", project);
 
-	arma::mat batch = createTraining("moby_ch1.txt");
-	batch.save("poet.training.dat", arma::arma_ascii);
+	arma::mat batch = createTraining("moby_ch1_small.txt");
+	batch.save("poet_small.training.dat", arma::arma_ascii);
 	
 	return 0;
 }