diff --git a/salt/modules/parted.py b/salt/modules/parted.py
index 72a6d3ce65..d2351e6683 100644
--- a/salt/modules/parted.py
+++ b/salt/modules/parted.py
@@ -93,7 +93,7 @@ def _validate_partition_boundary(boundary):
     '''
     try:
         for unit in VALID_UNITS:
-            if boundary.endswith(unit):
+            if str(boundary).endswith(unit):
                 return
         int(boundary)
     except Exception: