Update after checkout