diff --git a/glance/tests/unit/v2/test_images_resource.py b/glance/tests/unit/v2/test_images_resource.py
index c9b5b6b5de..da340253cc 100644
--- a/glance/tests/unit/v2/test_images_resource.py
+++ b/glance/tests/unit/v2/test_images_resource.py
@@ -3877,9 +3877,7 @@ class TestImagesController(base.IsolatedUnitTest):
             virtual_size=3072, extra_properties=props)
         self.controller._delete_encryption_key(request.context, image)
         # Make sure the encryption key is gone
-        # TODO(tkajinam): Replace Exception by ManagedObjectNotFoundError once
-        #                 castellan in u-c is bumped to 4.4.0
-        self.assertRaises(Exception,  # noqa
+        self.assertRaises(castellan_exception.ManagedObjectNotFoundError,
                           self.controller._key_manager.get,
                           request.context, fake_encryption_key)