check if component is not html

mui · May 1, 2024 · 1c20980 · 1c20980
1 parent a5f2b38
commit 1c20980
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/packages/pigment-css-react/src/styled.js b/packages/pigment-css-react/src/styled.js
@@ -123,14 +123,14 @@ export default function styled(tag, componentMeta = {}) {
         getVariantClasses(inProps, variants),
       );
 
-      if (shouldUseAs && !shouldForwardProp) {
-        // Reassign `shouldForwardProp` based on the incoming `as` prop
-        if (isHtmlTag(Component)) {
-          finalShouldForwardProp = isPropValid;
-        } else if (slot === 'Root' || slot === 'root') {
-          finalShouldForwardProp = rootShouldForwardProp;
-        } else {
-          finalShouldForwardProp = slotShouldForwardProp;
+      if (inProps.as && !shouldForwardProp) {
+        // Reassign `shouldForwardProp` if incoming `as` prop is a React component
+        if (!isHtmlTag(Component)) {
+          if (slot === 'Root' || slot === 'root') {
+            finalShouldForwardProp = rootShouldForwardProp;
+          } else {
+            finalShouldForwardProp = slotShouldForwardProp;
+          }
         }
       }