DRY change on test file