diff --git a/projects/testing-library/src/lib/models.ts b/projects/testing-library/src/lib/models.ts
index 1167925a..bfe01972 100644
--- a/projects/testing-library/src/lib/models.ts
+++ b/projects/testing-library/src/lib/models.ts
@@ -11,7 +11,7 @@ export interface RenderResult<ComponentType, WrapperType = ComponentType> extend
    * The containing DOM node of your rendered Angular Component.
    * This is a regular DOM node, so you can call container.querySelector etc. to inspect the children.
    */
-  container: HTMLElement;
+  container: Element;
   /**
    * @description
    * Prints out the component's DOM with syntax highlighting.
diff --git a/projects/testing-library/src/lib/testing-library.ts b/projects/testing-library/src/lib/testing-library.ts
index b347b1de..c5237fdb 100644
--- a/projects/testing-library/src/lib/testing-library.ts
+++ b/projects/testing-library/src/lib/testing-library.ts
@@ -273,7 +273,7 @@ async function waitForElementToBeRemovedWrapper<T>(
 ): Promise<void> {
   let cb;
   if (typeof callback !== 'function') {
-    const elements = (Array.isArray(callback) ? callback : [callback]) as HTMLElement[];
+    const elements = (Array.isArray(callback) ? callback : [callback]) as Element[];
     const getRemainingElements = elements.map((element) => {
       let parent = element.parentElement;
       while (parent.parentElement) {