diff --git a/tests/test_flavors.py b/tests/test_flavors.py
index 711a7fc..fe212b3 100644
--- a/tests/test_flavors.py
+++ b/tests/test_flavors.py
@@ -40,8 +40,8 @@ class TestFlavor:
         a = flavor.A
         b = flavor.B
 
-        assert type(a) == flavor
-        assert type(b) == flavor
+        assert type(a) is flavor
+        assert type(b) is flavor
         assert a.value == 1
         assert b.value == 2
         assert a is flavor.A
diff --git a/tests/test_stories.py b/tests/test_stories.py
index 78af850..d087b72 100644
--- a/tests/test_stories.py
+++ b/tests/test_stories.py
@@ -210,7 +210,7 @@ class TestStory:
 
         assert story.flavors is not flavors
         assert story.flavors == {flavor.A}
-        assert type(story.flavors) == set
+        assert type(story.flavors) is set
 
     def test_merge_without_parameters(self, story):
         """