diff --git a/src/plugin/hypo_info.h b/src/plugin/hypo_info.h
index 30f738c4..1b33ee00 100644
--- a/src/plugin/hypo_info.h
+++ b/src/plugin/hypo_info.h
@@ -23,6 +23,18 @@ struct HypoState
 
 typedef std::vector<HypoState> HypoStates;
 
+////////////////////////////////////////////////////////////////
+struct AmunInput : public HypoState
+{
+  AmunInput(const HypoState &hypoState)
+  :HypoState(hypoState)
+  {
+  }
+
+  Words phrase;
+};
+
+typedef std::vector<AmunInput> AmunInputs;
 
 }
 
diff --git a/src/plugin/nmt.cu b/src/plugin/nmt.cu
index 8b372b64..fc362a1c 100644
--- a/src/plugin/nmt.cu
+++ b/src/plugin/nmt.cu
@@ -55,7 +55,7 @@ HypoState MosesPlugin::SetSource(const std::vector<size_t>& words) {
   return ret;
 }
 
-HypoStates MosesPlugin::Score(const HypoStates &inputs)
+HypoStates MosesPlugin::Score(const AmunInputs &inputs)
 {
   HypoStates outputs(inputs.size());
 
diff --git a/src/plugin/nmt.h b/src/plugin/nmt.h
index 5ae39cc6..af8ad6a6 100644
--- a/src/plugin/nmt.h
+++ b/src/plugin/nmt.h
@@ -34,7 +34,7 @@ class MosesPlugin {
 
     HypoState SetSource(const std::vector<size_t>& words);
 
-    HypoStates Score(const HypoStates &inputs);
+    HypoStates Score(const AmunInputs &inputs);
 
   private:
     amunmt::God god_;