A small clean up.