diff --git a/vitrage/tests/unit/datasources/consistency/test_consistency_transformer.py b/vitrage/tests/unit/datasources/consistency/test_consistency_transformer.py
index 6482b38de..58f491681 100644
--- a/vitrage/tests/unit/datasources/consistency/test_consistency_transformer.py
+++ b/vitrage/tests/unit/datasources/consistency/test_consistency_transformer.py
@@ -64,7 +64,7 @@ class TestConsistencyTransformer(base.BaseTest):
             self._validate_consistency_vertex_props(vertex, event)
 
             neighbors = wrapper.neighbors
-            self.assertEqual(None, neighbors)
+            self.assertIsNone(neighbors)
 
             action = wrapper.action
             self.assertIn(action, self.actions)
@@ -88,7 +88,7 @@ class TestConsistencyTransformer(base.BaseTest):
             self._validate_consistency_vertex_props(vertex, event)
 
             neighbors = wrapper.neighbors
-            self.assertEqual(None, neighbors)
+            self.assertIsNone(neighbors)
 
             action = wrapper.action
             self.assertIn(action, self.actions)