diff --git a/config-models/aether-2.0.x/files/yang/onf-traffic-class.yang b/config-models/aether-2.0.x/files/yang/onf-traffic-class.yang
index f8658fb..5f1e021 100755
--- a/config-models/aether-2.0.x/files/yang/onf-traffic-class.yang
+++ b/config-models/aether-2.0.x/files/yang/onf-traffic-class.yang
@@ -5,7 +5,6 @@
 submodule onf-traffic-class {
   belongs-to onf-enterprise { prefix ent; }
 
-  import ietf-yang-types{ prefix yg; }
   import onf-aether-types{ prefix at; }
 
   organization "Open Networking Foundation.";
@@ -25,9 +24,7 @@
   }
 
   typedef traffic-class-id {
-        type yg:yang-identifier {
-            length 1..32;
-        }
+    type at:aether-identifier;
     description "The typedef for traffic-class-id";
   }