minor changes to testing