Small code style tidy up