diff --git a/.github/workflows/rust.yml b/.github/workflows/rust.yml
index d84d7f8..2657a81 100644
--- a/.github/workflows/rust.yml
+++ b/.github/workflows/rust.yml
@@ -13,10 +13,15 @@ env:
 jobs:
   build:
     runs-on: ${{matrix.os}}
+    env:
+      DISPLAY: ':99'
     strategy:
       fail-fast: false
       matrix: 
         os: [macos-latest, ubuntu-latest, windows-latest]
+        include:
+        - os: ubuntu-latest
+          headless: Xvfb :99 -screen 0 1024x768x24 > /dev/null 2>&1 &
 
     steps:
       - uses: actions/checkout@v3
@@ -30,6 +35,9 @@ jobs:
           packages: libxtst-dev libevdev-dev libxdo-dev
           version: 1.0
 
+      - name: Setup headless environment
+        run: ${{matrix.headless}}
+
       - name: Add components
         run: rustup component add clippy rustfmt
 
diff --git a/src/lib.rs b/src/lib.rs
index 0f19e06..0003e63 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -458,7 +458,6 @@ impl Frontend for Wish {
 mod tests {
     use crate::{Config, Wish};
     use afrim::frontend::Frontend;
-    use rstk::{self};
     use std::path::Path;
     use std::thread;
     use std::time::Duration;
@@ -487,8 +486,8 @@ mod tests {
         afrim_wish.display();
 
         // Test the geometry.
-        (0..800).for_each(|i| {
-            if i % 100 != 0 {
+        (0..100).for_each(|i| {
+            if i % 10 != 0 {
                 return;
             };
             let i = i as f64;
@@ -508,21 +507,5 @@ mod tests {
             Some(&("test".to_owned(), "123".to_owned(), "ok".to_owned()))
         );
         afrim_wish.display();
-
-        // Test error message.
-        {
-            let afrim_wish = afrim_wish.clone();
-            rstk::after(300, move || {
-                afrim_wish.raise_error("Intentional error", "For testing purpose");
-            });
-        }
-        {
-            let afrim_wish = afrim_wish.clone();
-            rstk::after(500, move || {
-                afrim_wish.destroy();
-            });
-        }
-
-        afrim_wish.listen();
     }
 }