Small improvement to ending.