diff --git a/shark/iree_utils/_common.py b/shark/iree_utils/_common.py
index 26dcf7c5b5..c58405b46e 100644
--- a/shark/iree_utils/_common.py
+++ b/shark/iree_utils/_common.py
@@ -91,7 +91,7 @@ def iree_target_map(device):
     "cpu-task": "llvm-cpu",
     "cpu-sync": "llvm-cpu",
     "cuda": "cuda",
-    "vulkan": "vulkan",
+    "vulkan": "vulkan-spirv",
     "metal": "metal",
     "rocm": "rocm",
     "intel-gpu": "opencl-spirv",
@@ -122,9 +122,7 @@ def check_device_drivers(device):
         )
         return True
     except RuntimeError as re:
-        print(
-            f"[ERR] Failed to get driver for {device} with error:\n{repr(re)}"
-        )
+        print(f"[ERR] Failed to get driver for {device} with error:\n{repr(re)}")
         return True
 
     # Unknown device. We assume drivers are installed.