diff --git a/src/test/java/io/github/sridharbandi/util/A11yTest.java b/src/test/java/io/github/sridharbandi/util/A11yTest.java
index eacb908..9e4cc00 100644
--- a/src/test/java/io/github/sridharbandi/util/A11yTest.java
+++ b/src/test/java/io/github/sridharbandi/util/A11yTest.java
@@ -1,7 +1,6 @@
 package io.github.sridharbandi.util;
 
 import com.microsoft.playwright.Page;
-import com.microsoft.playwright.options.LoadState;
 import freemarker.template.Template;
 import io.github.sridharbandi.a11y.Engine;
 import io.github.sridharbandi.a11y.HTMLCS;
@@ -26,7 +25,6 @@
 import java.util.Objects;
 
 import static org.junit.jupiter.api.Assertions.assertTrue;
-import static org.mockito.Mockito.doNothing;
 
 @ExtendWith(MockitoExtension.class)
 public class A11yTest {
@@ -44,8 +42,7 @@ public void setUp() {
     }
 
     @Test
-    public void testExecute() throws Exception {
-        doNothing().when(page).waitForLoadState(LoadState.DOMCONTENTLOADED);
+    public void testHtmlCsExecute() throws Exception {
         Params params = new Params();
         params.setStandard(HTMLCS.WCAG2AA.name());
         a11y.execute(Engine.HTMLCS, params);
@@ -53,6 +50,15 @@ public void testExecute() throws Exception {
         assertTrue(FileUtils.deleteQuietly(Objects.requireNonNull(new File("./target/java-a11y/htmlcs/json").listFiles())[0]));
     }
 
+    @Test
+    public void testHtmlAxeExecute() throws Exception {
+        Params params = new Params();
+        params.setPageTitle("Page Title");
+        a11y.execute(Engine.AXE, params);
+        a11y.jsonReports(Engine.AXE, io.github.sridharbandi.modal.axe.Issues.class);
+        assertTrue(FileUtils.deleteQuietly(Objects.requireNonNull(new File("./target/java-a11y/axe/json").listFiles())[0]));
+    }
+
     @Test
     public void testSave() throws IOException {
         Template template = ftlConfig.getTemplate("test.ftl");