Minor extension to testing