1 files changed, 6 insertions, 2 deletions
diff --git a/src/include/fst/state-map.h b/src/include/fst/state-map.h
index ace4a3c..454db24 100644
--- a/src/include/fst/state-map.h
+++ b/src/include/fst/state-map.h
@@ -295,6 +295,10 @@ class StateMapFstImpl : public CacheImpl<B> {
     SetArcs(s);
   }
 
+  const Fst<A> &GetFst() const {
+    return *fst_;
+  }
+
  private:
   void Init() {
     SetType("statemap");
@@ -364,10 +368,10 @@ class StateMapFst : public ImplToFst< StateMapFstImpl<A, B, C> > {
     GetImpl()->InitArcIterator(s, data);
   }
 
- private:
-  // Makes visible to friends.
+ protected:
   Impl *GetImpl() const { return ImplToFst<Impl>::GetImpl(); }
 
+ private:
   void operator=(const StateMapFst<A, B, C> &fst);  // disallow
 };