make sure to handle the cases where we are done.