tiny code tidy-up (nothing major)