Could you move this to testing utils