removing more axis='sample'