diff --git a/python/Scripts/creatematerial.py b/python/Scripts/creatematerial.py
index d9781c357e..2b536720d5 100644
--- a/python/Scripts/creatematerial.py
+++ b/python/Scripts/creatematerial.py
@@ -177,7 +177,7 @@ def buildDocument(textureFiles, mtlxFile, shadingModel, colorspace, useTiledImag
         imageNode = nodeGraph.addNode(imageNodeCategory, textureName, inputType)
 
         # Set color space.
-        if 'color' in inputType.lower():
+        if shaderInput.isColorType():
             imageNode.setColorSpace(colorspace)
 
         # Set file path.