diff --git a/yang/frr-zebra.yang b/yang/frr-zebra.yang
index 2fc9e5f1bc42..ac955384354c 100644
--- a/yang/frr-zebra.yang
+++ b/yang/frr-zebra.yang
@@ -634,9 +634,7 @@ module frr-zebra {
           list route-entry {
             key "protocol";
             leaf protocol {
-               type frr-route-types:frr-route-types-v4;
-               //TODO: Use unified route types done in PR 5183 when it is merged.
-               //type frr-route-types:frr-route-types;
+               type frr-route-types:frr-route-types;
                description
                   "The protocol owning the route.";
             }