diff --git a/inventree_bulk_plugin/tests/integration/test_bulkcreate_objects.py b/inventree_bulk_plugin/tests/integration/test_bulkcreate_objects.py
index d1dec3c..4367dd1 100644
--- a/inventree_bulk_plugin/tests/integration/test_bulkcreate_objects.py
+++ b/inventree_bulk_plugin/tests/integration/test_bulkcreate_objects.py
@@ -709,6 +709,5 @@ def test_get_context(self):
         r.data = {"template": {"output": {"generate": {"variant_of": str(part.pk)}}}}
         obj = PartBulkCreateObject(r)
         ctx = obj.get_context()
-        self.assertDictContainsSubset(
-            {"name": "Test part", "description": "Test part desc", "is_template": True}, ctx["gen"])
+        self.assertEqual(ctx["gen"] | {"name": "Test part", "description": "Test part desc", "is_template": True}, ctx["gen"])
         self.assertEqual(obj.parent, part)