YouTube Music is yet another of Google’s forays into streaming music. It can also use a huge chunk of your mobile data. Now, the app finally allows you to limit your streaming to Wi-Fi only.

The latest update adds a toggle that will only allow you to stream whenever your phone is connected to a Wi-Fi network. Since YouTube Music can stream videos as well as songs, it can easily blow through your data limit quickly. Of course, the previous options to limit data are also available: you can enable limited data use by only streaming HD videos when you’re over Wi-Fi (seen above). Additionally, there’s the usual toggle switch in the app that switches between music-only and video modes. All are handy options for a service that has the potential to use too much data in a short amount of time.