1 files changed, 6 insertions, 2 deletions
diff --git a/unitest-xmlrpc.py b/unitest-xmlrpc.py
index d1e56d4f..4d14ce13 100755
--- a/unitest-xmlrpc.py
+++ b/unitest-xmlrpc.py
@@ -27,9 +27,13 @@ import time
 import unittest
 import shlex
 import subprocess
-import xmlrpclib
 import json
 import types
+try:
+    from xmlrpc.client import ServerProxy
+except ImportError:  
+    # Python 2
+    from xmlrpclib import ServerProxy
 
 from glances.core.glances_globals import (
     appname,
@@ -68,7 +72,7 @@ from glances.core.glances_stats import GlancesStats
 stats = GlancesStats()
 
 # Init the XML/RCP client
-client = xmlrpclib.ServerProxy(URL)
+client = ServerProxy(URL)
 
 # Unitest class
 # ==============