Fix ending bug.