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.

WiFi doesn't connect to saved network with screen off

Hi, I'm on 5.1.1 latest CM 12.1 nightly. My problem is in the title. I've updated my modem to the latest and still no go. As soon as I turn on the screen, it connects within 3 seconds. How do I get it to scan/connect with screen off? Thanks.

Edit* It works fine on Touchwiz.