Small cleanup.
Fix.