diff --git a/tests/unit/test_steps.py b/tests/unit/test_steps.py
index 29b8c28..686432d 100644
--- a/tests/unit/test_steps.py
+++ b/tests/unit/test_steps.py
@@ -3,4 +3,4 @@
 
 def test_smoke():
     """Smoke test of Step registration."""
-    assert type(step_types) == StepTypes
+    assert isinstance(step_types, StepTypes)