Tidy up the last PR.