Small clean up.
second commit