Fix silly test breakage