Minor tweaks to test code based on pull request feedback