diff --git a/sky130/custom/scripts/fix_device_models.py b/sky130/custom/scripts/fix_device_models.py
index f9a258e..2c746b5 100755
--- a/sky130/custom/scripts/fix_device_models.py
+++ b/sky130/custom/scripts/fix_device_models.py
@@ -57,7 +57,8 @@
             fixedlines.append(fline)
             modified = True
         elif smatch:
-            fline = re.sub('short', 'sky130_fd_pr__res_generic_po', line)
+            fline = re.sub(' VNB short', ' sky130_fd_pr__res_generic_po', line)
+            fline = re.sub('short', 'sky130_fd_pr__res_generic_po', fline)
             fline = re.sub('^X', 'R', fline)
             fixedlines.append(fline)
             modified = True