Make the partition icon more general