Small clean up.