Ref minor tidy up