diff --git a/pkgs/sdk-pkg/src/genie/libs/sdk/apis/iosxe/bgp/configure.py b/pkgs/sdk-pkg/src/genie/libs/sdk/apis/iosxe/bgp/configure.py
index 0c756aa0e..9da1adc1c 100755
--- a/pkgs/sdk-pkg/src/genie/libs/sdk/apis/iosxe/bgp/configure.py
+++ b/pkgs/sdk-pkg/src/genie/libs/sdk/apis/iosxe/bgp/configure.py
@@ -606,9 +606,13 @@ def configure_shut_bgp_neighbors(
 
     if neighbors:
         if noshut:
+            if vrf:
+                cmd += "address-family {} vrf {}\n".format(address_family, vrf)
             for neighbor in neighbors:
                 cmd += "no neighbor {} shutdown".format(neighbor)
         else:
+            if vrf:
+                cmd += "address-family {} vrf {}\n".format(address_family, vrf)
             for neighbor in neighbors:
                 cmd += "neighbor {} shutdown".format(neighbor)
         try: