dev: fix restart bug