1 files changed, 4 insertions, 4 deletions
diff --git a/ebooks.cpp b/ebooks.cpp
index 8b567ed..52247da 100644
--- a/ebooks.cpp
+++ b/ebooks.cpp

@@ -64,14 +64,14 @@ int main(int argc, char** args)
  rawr kgramstats;
  kgramstats.addCorpus(corpus);
  kgramstats.compile(4);
-  kgramstats.setTransformCallback([&] (std::string canonical, std::string) {
+  kgramstats.setTransformCallback([&] (std::string, std::string form) {
-    size_t pos = canonical.find("$name$");
+    size_t pos = form.find("$name$");
    if (pos != std::string::npos)
    {
-      canonical.replace(pos, 6, fv_names[rand() % fv_names.size()]);
+      form.replace(pos, 6, fv_names[rand() % fv_names.size()]);
    }
    
-    return canonical;
+    return form;
  });
  
  std::mutex stats_mutex;

diff --git a/ebooks.cpp b/ebooks.cpp index 8b567ed..52247da 100644 --- a/ebooks.cpp +++ b/ebooks.cpp
@@ -64,14 +64,14 @@ int main(int argc, char** args)
64	rawr kgramstats;	64	rawr kgramstats;
65	kgramstats.addCorpus(corpus);	65	kgramstats.addCorpus(corpus);
66	kgramstats.compile(4);	66	kgramstats.compile(4);
67	kgramstats.setTransformCallback([&] (std::string canonical, std::string) {	67	kgramstats.setTransformCallback([&] (std::string, std::string form) {
68	size_t pos = canonical.find("$name$");	68	size_t pos = form.find("$name$");
69	if (pos != std::string::npos)	69	if (pos != std::string::npos)
70	{	70	{
71	canonical.replace(pos, 6, fv_names[rand() % fv_names.size()]);	71	form.replace(pos, 6, fv_names[rand() % fv_names.size()]);
72	}	72	}
73		73
74	return canonical;	74	return form;
75	});	75	});
76		76
77	std::mutex stats_mutex;	77	std::mutex stats_mutex;