From 3ec5ff5ec9c972060219a8613070ccaf5b6f5c4f Mon Sep 17 00:00:00 2001
From: Mike Hiretsky <mh@calculate.ru>
Date: Fri, 22 Jul 2011 14:35:25 +0400
Subject: [PATCH] Fix change part id.

---
 pym/cl_distr.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pym/cl_distr.py b/pym/cl_distr.py
index bfaf3de..97f110e 100644
--- a/pym/cl_distr.py
+++ b/pym/cl_distr.py
@@ -26,7 +26,7 @@ import operator
 from cl_utils import runOsCommand,isMount,removeDir,typeFile,pathJoin, \
                     process,getRunCommands,getTupleVersion,cmpVersion, \
                     detectDeviceForPartition, getProgPath,listDirectory, \
-                    checkUtils,STDOUT,getUdevDeviceInfo
+                    checkUtils,STDOUT,getUdevDeviceInfo, countPartitions
 from cl_vars_share import varsShare
 from shutil import copyfile,copytree
 from cl_template import _terms
@@ -936,7 +936,7 @@ class PartitionDistributive(Distributive):
         elif deviceName == "":
             return True
         fdiskProg, gdiskProg = checkUtils('/sbin/fdisk','/usr/sbin/gdisk')
-        partitionNumber = getUdevDeviceInfo(dev).get('ID_PART_ENTRY_NUMBER','')
+        partitionNumber = getUdevDeviceInfo(name=dev).get('ID_PART_ENTRY_NUMBER','')
         devicePartitionCount = countPartitions(deviceName)
         if deviceName and not partitionNumber:
             raise DistributiveError(