Small fix when restarting simulations