Minor file tidy up, missed from previous commit