Fixed a single restart bug