diff --git a/src/main/java/org/onosproject/segmentrouting/PolicyHandler.java b/src/main/java/org/onosproject/segmentrouting/PolicyHandler.java
index 9a2a891..e773e10 100644
--- a/src/main/java/org/onosproject/segmentrouting/PolicyHandler.java
+++ b/src/main/java/org/onosproject/segmentrouting/PolicyHandler.java
@@ -18,6 +18,7 @@
 
 import org.onlab.packet.Ethernet;
 import org.onlab.packet.IpPrefix;
+import org.onlab.packet.TpPort;
 import org.onosproject.cli.net.IpProtocol;
 import org.onosproject.core.ApplicationId;
 import org.onosproject.net.DeviceId;
@@ -193,17 +194,17 @@
             tsb.matchIPProtocol(ipProto.byteValue());
             if (IpProtocol.valueOf(policy.ipProto()).equals(IpProtocol.TCP)) {
                 if (policy.srcPort() != 0) {
-                    tsb.matchTcpSrc(policy.srcPort());
+                    tsb.matchTcpSrc(TpPort.tpPort(policy.srcPort()));
                 }
                 if (policy.dstPort() != 0) {
-                    tsb.matchTcpDst(policy.dstPort());
+                    tsb.matchTcpDst(TpPort.tpPort(policy.dstPort()));
                 }
             } else if (IpProtocol.valueOf(policy.ipProto()).equals(IpProtocol.UDP)) {
                 if (policy.srcPort() != 0) {
-                    tsb.matchUdpSrc(policy.srcPort());
+                    tsb.matchUdpSrc(TpPort.tpPort(policy.srcPort()));
                 }
                 if (policy.dstPort() != 0) {
-                    tsb.matchUdpDst(policy.dstPort());
+                    tsb.matchUdpDst(TpPort.tpPort(policy.dstPort()));
                 }
             }
         }