diff --git a/p4app/controller.py b/p4app/controller.py
index adbd0f0..fcb1fb1 100644
--- a/p4app/controller.py
+++ b/p4app/controller.py
@@ -7,8 +7,9 @@ from p4utils.utils.topology import Topology
 from p4utils.utils.sswitch_API import SimpleSwitchAPI
 
 from scapy.all import sniff, get_if_list, Ether, get_if_hwaddr, sendp
-from scapy.all import IP, Raw, IPv6, TCP, TCP_client
-from scapy.all import Ether, sniff, Packet, BitField, IntEnumField, ShortField
+from scapy.all import IP, Raw, IPv6, TCP, TCP_client, Ether
+from scapy.all import sniff
+from scapy.all import Packet, BitField, IntEnumField, ShortField, XShortEnumField
 from scapy.all import ICMPv6ND_NS
 from scapy.data import ETHER_TYPES