Slight fix to testing