diff --git a/mmseg/models/decode_heads/knet_head.py b/mmseg/models/decode_heads/knet_head.py
index 46de21434..74f371598 100644
--- a/mmseg/models/decode_heads/knet_head.py
+++ b/mmseg/models/decode_heads/knet_head.py
@@ -414,7 +414,7 @@ class IterativeDecodeHead(BaseDecodeHead):
         super(BaseDecodeHead, self).__init__(**kwargs)
         assert num_stages == len(kernel_update_head)
         self.num_stages = num_stages
-        self.kernel_generate_head = (kernel_generate_head)
+        self.kernel_generate_head = MODELS.build(kernel_generate_head)
         self.kernel_update_head = nn.ModuleList()
         self.align_corners = self.kernel_generate_head.align_corners
         self.num_classes = self.kernel_generate_head.num_classes