diff --git a/tests/Settings/TestGlobalStack.py b/tests/Settings/TestGlobalStack.py
index f769384c16..ae94089246 100644
--- a/tests/Settings/TestGlobalStack.py
+++ b/tests/Settings/TestGlobalStack.py
@@ -542,5 +542,5 @@ def test_setVariantByIdExists(global_stack, container_registry):
 
 ##  Tests setting variants by specifying an ID of a variant that doesn't exist.
 def test_setVariantByIdDoesntExist(global_stack):
-    with pytest.raises(KeyError):
-        global_stack.setVariantById("some_variant") #Container registry is empty now.
\ No newline at end of file
+    with pytest.raises(InvalidContainerError):
+        global_stack.setVariantById("some_variant") #Container registry is empty now.