Can the music client prevent wifi for turning off when it's downloading songs?

[Guilherme Salgado]

Always when I'm listening to an album which hasn't been cached yet, the music client stops playing at some point halfway through. When I wake up the phone I can see it's disconnected from the wifi, and then I soon as it connects again the music client resumes playing. I'm pretty sure this is because Android will turn the wifi off when the phone goes to sleep (although I'm not sure what that means in terms of time from when the screen was turned off), so I wonder if it wouldn't be possible to have the U1 music client tell Android not to shut down wifi until it has finished downloading the songs that are queued?