CORD-1389: xproto extension to support policies
Change-Id: I5d6c11107d5bc9cd82e41c4a8e6a378d22d7fb61
diff --git a/lib/xos-genx/xosgenx/proto2xproto.py b/lib/xos-genx/xosgenx/proto2xproto.py
index f05c2bc..3723e7c 100644
--- a/lib/xos-genx/xosgenx/proto2xproto.py
+++ b/lib/xos-genx/xosgenx/proto2xproto.py
@@ -1,4 +1,5 @@
import plyxproto.model as m
+from plyxproto.helpers import Visitor
import pdb
import argparse
import plyxproto.parser as plyxproto
@@ -47,7 +48,7 @@
except:
return obj
-class Proto2XProto(m.Visitor):
+class Proto2XProto(Visitor):
def __init__(self):
super(Proto2XProto, self).__init__()