Use MediaState enum