How do I keep an element's shape in CSS?

.square {
  width: 150px;
  aspect-ratio: 1;
}