Small fix.
Tidy up.