diff --git a/tests/modules/test_example.py b/tests/modules/test_example.py
index 3615560..a8829af 100644
--- a/tests/modules/test_example.py
+++ b/tests/modules/test_example.py
@@ -1,12 +1,13 @@
 # -*- coding: utf-8 -*-
 
-import pytest
+import unittest
 
 from rookify.modules.example.main import ExampleHandler
 from rookify.modules.machine import Machine
 from rookify.modules.module import ModuleException
 
 
-def test_preflight() -> None:
-    with pytest.raises(ModuleException):
-        ExampleHandler(Machine(), {}).preflight()
+class TestExampleHandler(unittest.TestCase):
+    def test_preflight(self) -> None:
+        with self.assertRaises(ModuleException):
+            ExampleHandler(Machine(), {}).preflight()