Minor update to testing