1 files changed, 13 insertions, 2 deletions
diff --git a/source/slang/slang-check-modifier.cpp b/source/slang/slang-check-modifier.cpp
index 2bc914a65..ce80d0002 100644
--- a/source/slang/slang-check-modifier.cpp
+++ b/source/slang/slang-check-modifier.cpp
@@ -591,9 +591,20 @@ namespace Slang
             }
 
             ImageFormat format = ImageFormat::unknown;
-            if(!findImageFormatByName(formatName.getBuffer(), &format))
+
+            if (attr->keywordName->text.getUnownedSlice() == toSlice("image"))
+            {
+                if(!findImageFormatByName(formatName.getUnownedSlice(), &format))
+                {
+                    getSink()->diagnose(attr->args[0], Diagnostics::unknownImageFormatName, formatName);
+                }
+            }
+            else
             {
-                getSink()->diagnose(attr->args[0], Diagnostics::unknownImageFormatName, formatName);
+                if (!findVkImageFormatByName(formatName.getUnownedSlice(), &format))
+                {
+                    getSink()->diagnose(attr->args[0], Diagnostics::unknownImageFormatName, formatName);
+                }
             }
 
             formatAttr->format = format;