Fix tests for new psutil