diff --git a/neutron_classifier/db/api.py b/neutron_classifier/db/api.py
index dd48a70..bf2d4e3 100644
--- a/neutron_classifier/db/api.py
+++ b/neutron_classifier/db/api.py
@@ -65,8 +65,8 @@ def convert_security_group_rule_to_classifier(context, sgr, group):
 
     # Ports
     cl2 = models.TransportClassifier(
-        dst_port_range_min=sgr['port_range_min'],
-        dst_port_range_max=sgr['port_range_max'])
+        destination_port_range_min=sgr['port_range_min'],
+        destination_port_range_max=sgr['port_range_max'])
 
     # Direction
     cl3 = models.DirectionClassifier(
diff --git a/neutron_classifier/db/models.py b/neutron_classifier/db/models.py
index 0aaf7fa..e4d68ae 100644
--- a/neutron_classifier/db/models.py
+++ b/neutron_classifier/db/models.py
@@ -153,11 +153,15 @@ class TransportClassifier(Classifier):
     destination_port_range_max = sa.Column(sa.Integer)
     destination_port_range_min = sa.Column(sa.Integer)
 
-    def __init__(self, dst_port_range_min=None,
-                 dst_port_range_max=None):
+    def __init__(self, source_port_range_min=None,
+                 source_port_range_max=None,
+                 destination_port_range_min=None,
+                 destination_port_range_max=None):
         super(TransportClassifier, self).__init__()
-        self.destination_port_range_min = dst_port_range_min
-        self.destination_port_range_max = dst_port_range_max
+        self.destination_port_range_min = destination_port_range_min
+        self.destination_port_range_max = destination_port_range_max
+        self.source_port_range_min = source_port_range_min
+        self.source_port_range_max = source_port_range_max
 
 
 class VlanClassifier(Classifier):