diff --git a/container/container_contract.go b/container/container_contract.go
index 244805f..b2a319b 100644
--- a/container/container_contract.go
+++ b/container/container_contract.go
@@ -92,6 +92,8 @@ func _deploy(data interface{}, isUpdate bool) {
 	nnsRoot := args[5].(string)
 
 	if isUpdate {
+		storage.Put(ctx, nnsContractKey, addrNNS)
+		storage.Put(ctx, nnsRootKey, nnsRoot)
 		registerNiceNameTLD(addrNNS, nnsRoot)
 		return
 	}