Fix: fix assign hang forever when closing

[confluent-cla-assistant[bot]]

:tada: All Contributor License Agreements have been signed. Ready to merge.
:white_check_mark: aiquestion
_{Please push an empty commit if you would like to re-run the checks to verify CLA status for all contributors.}