1 files changed, 5 insertions, 6 deletions
diff --git a/klm/search/final.hh b/klm/search/final.hh
index 823b8c1a..1b3092ac 100644
--- a/klm/search/final.hh
+++ b/klm/search/final.hh
@@ -2,35 +2,34 @@
 #define SEARCH_FINAL__
 
 #include "search/arity.hh"
+#include "search/note.hh"
 #include "search/types.hh"
 
 #include <boost/array.hpp>
 
 namespace search {
 
-class Edge;
-
 class Final {
   public:
     typedef boost::array<const Final*, search::kMaxArity> ChildArray;
 
-    void Reset(Score bound, const Edge &from, const Final &left, const Final &right) {
+    void Reset(Score bound, Note note, const Final &left, const Final &right) {
       bound_ = bound;
-      from_ = &from;
+      note_ = note;
       children_[0] = &left;
       children_[1] = &right;
     }
 
     const ChildArray &Children() const { return children_; }
 
-    const Edge &From() const { return *from_; }
+    Note GetNote() const { return note_; }
 
     Score Bound() const { return bound_; }
 
   private:
     Score bound_;
 
-    const Edge *from_;
+    Note note_;
 
     ChildArray children_;
 };