make setting for as a element work

umbraco · May 24, 2024 · dee0d8c · dee0d8c
1 parent e159c43
commit dee0d8c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/packages/uui-form-validation-message/lib/uui-form-validation-message.element.ts b/packages/uui-form-validation-message/lib/uui-form-validation-message.element.ts
@@ -27,12 +27,14 @@ export class UUIFormValidationMessageElement extends LitElement {
     return this._for;
   }
   public set for(value: HTMLElement | string | null) {
-    let queriedElement = null;
+    let element = null;
     if (typeof value === 'string') {
       const scope = this.getRootNode();
-      queriedElement = (scope as DocumentFragment)?.getElementById(value);
+      element = (scope as DocumentFragment)?.getElementById(value);
+    } else if (value instanceof HTMLElement) {
+      element = value;
     }
-    const newScope = queriedElement || this;
+    const newScope = element ?? this;
     const oldScope = this._for;
 
     if (oldScope === newScope) {