Fix the end function bug.