UPDATE: Following dylanthecat's suggestion I rebooted the router and that seems to have fixed the problem. The WiFi goes to sleep when the screen goes off and wakes back up when I reopen it. Thanks