Fix restarting

Signed-off-by: Vlad <vg@vladgh.com>