Minor clean up.
Minor clean up.
Small fix.