diff --git a/ofagent/converter.py b/ofagent/converter.py
index 38ceb03..7e26681 100755
--- a/ofagent/converter.py
+++ b/ofagent/converter.py
@@ -101,7 +101,7 @@
             of13.oxm.ip_proto(value=ofb_field['ip_proto']))
 
     elif field_type == pb2.OFPXMT_OFB_VLAN_VID:
-        if ofb_field.get('vlan_vid') == RESERVED_TRANSPARENT_VLAN and ofb_field['vlan_vid_mask'] == RESERVED_TRANSPARENT_VLAN:
+        if ofb_field.get('vlan_vid') == RESERVED_TRANSPARENT_VLAN and 'vlan_vid_mask' in ofb_field and ofb_field['vlan_vid_mask'] == RESERVED_TRANSPARENT_VLAN:
            return (
                of13.oxm.vlan_vid_masked(value=ofb_field['vlan_vid_mask'], value_mask=ofb_field['vlan_vid_mask']))
         else:
diff --git a/ofagent/of_connection.py b/ofagent/of_connection.py
index 8930472..a941111 100755
--- a/ofagent/of_connection.py
+++ b/ofagent/of_connection.py
@@ -98,7 +98,7 @@
         :return: None
         """
         assert self.connected
-        
+
         if msg.xid is None:
             msg.xid = self._gen_xid()
         buf = msg.pack()