1 files changed, 6 insertions, 2 deletions
diff --git a/python/src/kbest.pxd b/python/src/kbest.pxd
index e339714a..cef42bd3 100644
--- a/python/src/kbest.pxd
+++ b/python/src/kbest.pxd
@@ -1,16 +1,20 @@
 from libcpp.vector cimport vector
 from utils cimport WordID
-cimport hypergraph
+from hypergraph cimport Hypergraph
 
 cdef extern from "decoder/viterbi.h":
     cdef cppclass ESentenceTraversal:
         pass
     cdef cppclass ETreeTraversal:
         pass
+    cdef cppclass FTreeTraversal:
+        pass
+    cdef cppclass FeatureVectorTraversal:
+        pass
 
 cdef extern from "decoder/kbest.h" namespace "KBest":
     cdef cppclass KBestDerivations[T, Traversal]:
         cppclass Derivation:
             T _yield "yield"
-        KBestDerivations(hypergraph.Hypergraph hg, unsigned k)
+        KBestDerivations(Hypergraph& hg, unsigned k)
         Derivation* LazyKthBest(unsigned v, unsigned k)