Problemas com o WI-FI.

E ae galerinha, tenho um Moto G 1a Geração e ontem fi a atualização do Android 4.4.4 para o 5.0.2, só que percebi que meu aparelho quando entra em modo de inatividade ele desliga o WI-FI e os aplicativos que dependem da internet como o WhatsApp por exemplo não funcionam e as notificações só recebo quando ativo a tela e entro nos apps. O Android tem a função de escolher o modo de funcionamento do WI-FI durante a inatividade e eu deixei para o WI-FI sempre ficar ativo, mas mesmo configurado assim ele se auto desliga ao bloquear a tela. Gostaria de saber se alguem enfrenta o mesmo problema e se podem me dar um auxílio referente a esse problema.