Fix for wifi sleep issue