diff --git a/MaterialLib/MPL/Properties/VapourDiffusion/VapourDiffusionFEBEX.h b/MaterialLib/MPL/Properties/VapourDiffusion/VapourDiffusionFEBEX.h
index 0711b1b11ba..a38206d851a 100644
--- a/MaterialLib/MPL/Properties/VapourDiffusion/VapourDiffusionFEBEX.h
+++ b/MaterialLib/MPL/Properties/VapourDiffusion/VapourDiffusionFEBEX.h
@@ -63,11 +63,12 @@ class VapourDiffusionFEBEX final : public Property
 
     void checkScale() const override
     {
-        if (!std::holds_alternative<Phase*>(scale_))
+        if (!(std::holds_alternative<Phase*>(scale_) ||
+              std::holds_alternative<Component*>(scale_)))
         {
             OGS_FATAL(
                 "The property 'VapourDiffusionFEBEX' is "
-                "implemented on the 'phase' scale only.");
+                "implemented on the 'phase' and 'component' scale only.");
         }
     }
 
diff --git a/MaterialLib/MPL/Properties/VapourDiffusion/VapourDiffusionPMQ.h b/MaterialLib/MPL/Properties/VapourDiffusion/VapourDiffusionPMQ.h
index 692dd1edef5..db0114ecb5b 100644
--- a/MaterialLib/MPL/Properties/VapourDiffusion/VapourDiffusionPMQ.h
+++ b/MaterialLib/MPL/Properties/VapourDiffusion/VapourDiffusionPMQ.h
@@ -70,11 +70,12 @@ class VapourDiffusionPMQ final : public Property
 
     void checkScale() const override
     {
-        if (!std::holds_alternative<Phase*>(scale_))
+        if (!(std::holds_alternative<Phase*>(scale_) ||
+              std::holds_alternative<Component*>(scale_)))
         {
             OGS_FATAL(
                 "The property 'VapourDiffusionPMQ' is "
-                "implemented on the 'phase' scale only.");
+                "implemented on the 'phase' and 'component' scale only.");
         }
     }