Code tidy up #369