correct creation of input paths for lattices

1 files changed, 8 insertions, 1 deletions

diff --git a/moses/WordLattice.cpp b/moses/WordLattice.cpp
index fd2d7a831..d6e6e0c54 100644
--- a/moses/WordLattice.cpp
+++ b/moses/WordLattice.cpp
@@ -210,7 +210,14 @@ WordLattice::CreateTranslationOptionCollection() const
 {
   size_t maxNoTransOptPerCoverage = StaticData::Instance().GetMaxNoTransOptPerCoverage();
   float translationOptionThreshold = StaticData::Instance().GetTranslationOptionThreshold();
-  TranslationOptionCollection *rv= new TranslationOptionCollectionLattice(*this, maxNoTransOptPerCoverage, translationOptionThreshold);
+
+  TranslationOptionCollection *rv = NULL;
+  if (StaticData::Instance().GetUseLegacyPT()) {
+    rv = new TranslationOptionCollectionConfusionNet(*this, maxNoTransOptPerCoverage, translationOptionThreshold);
+  }
+  else {
+	rv = new TranslationOptionCollectionLattice(*this, maxNoTransOptPerCoverage, translationOptionThreshold);
+  }
   CHECK(rv);
   return rv;
 }