WiFi doesn't connect automatically with screen off

Hi there. It connects fine as soon as I turn the screen on. Not sure what the problem is. I believe it's a 5.1.1 issue.

http://forum.xda-developers.com/goog...issue-t3059452

That link has the same issue and they couldn't figure it out either.