If the problem goes away when switching off WiFi, then it's most likely that the WiFi router doesn't support power management (or doesn't have it enabled).