Small code tidy up.