diff --git a/projects/ngx-highlightjs/src/lib/highlight.spec.ts b/projects/ngx-highlightjs/src/lib/highlight.spec.ts
index a49cb97..9d4cb3f 100644
--- a/projects/ngx-highlightjs/src/lib/highlight.spec.ts
+++ b/projects/ngx-highlightjs/src/lib/highlight.spec.ts
@@ -33,7 +33,7 @@ describe('Highlight Directive', () => {
   const testJsCode = 'console.log("test")';
   const testHtmlCode = '<div class=&quot;my-class&quot;></div>';
 
-  beforeEach(() => {
+  beforeEach(waitForAsync(() => {
     TestBed.configureTestingModule({
       imports: [Highlight, TestHighlightComponent],
       providers: [
@@ -42,7 +42,9 @@ describe('Highlight Directive', () => {
       ]
     }).compileComponents();
     loader = TestBed.inject(HighlightLoader);
+  }));
 
+  beforeEach(() => {
     fixture = TestBed.createComponent(TestHighlightComponent);
     component = fixture.componentInstance;
     directiveElement = fixture.debugElement.query(By.directive(Highlight));