[PATCH master 2/7] Use GetClassFromIpFamily in ActivateMasterIp

[Andrea Spadaccini]

Signed-off-by: Andrea Spadaccini <spadac...@google.com>
---
 lib/backend.py |    4 +---
 1 files changed, 1 insertions(+), 3 deletions(-)

diff --git a/lib/backend.py b/lib/backend.py
index 8205381..8c77824 100644
--- a/lib/backend.py
+++ b/lib/backend.py
@@ -265,9 +265,7 @@ def ActivateMasterIp():
       err_msg = "Someone else has the master ip, not activating"
       logging.error(err_msg)
   else:
-    ipcls = netutils.IP4Address
-    if family == netutils.IP6Address.family:
-      ipcls = netutils.IP6Address
+    ipcls = netutils.IPAddress.GetClassFromIpFamily(ip_family)
 
     result = utils.RunCmd([constants.IP_COMMAND_PATH, "address", "add",
                            "%s/%s" % (master_ip, master_netmask),
-- 
1.7.3.1