Modified the restart test