Added error for restarts with small n_batches