Fix contract upgrade issue

[lovel8]

Describe the Bug

Fix contract upgrade issue

Repro Steps

1. Run '...'
2. Do '...'
3. See error '...'
   ...

Logging Information

No response

Additional context

No response