How do I check which team the entity is in 1.20.1?

An example of how I did it. From above, I checked whether the entity command contains an "entity command". Below, I used a block of logic. Thank you if you answer🤗.