Update for end case