Better attention to closing states