Transmission configuration indication (TCI) states may be used to indicate beam switching for data beams and/or control beams. A total set of TCI states may be divided into subsets of TCI states. TCI states conveyed via downlink control information (DCI) may thus act as beam switching indications depending on which subset a conveyed TCI state resides. A base station may transmit an indication that such DCI based control beam switching is enabled. Based on whether TCI state (e.g., included in DCI) belongs to a first subset of TCI states or a second subset of TCI states, a wireless device (e.g., that receives the DCI) may trigger a beam switching operation. Beam switching operations may include data beam switching or both data beam switching and control beam switching. The UE may then receive a downlink control channel and a downlink data channel based on the triggered beam switching operation.