restart bug fix