diff --git a/dislocker_slash.py b/dislocker_slash.py
index 4f7893b..262cacb 100644
--- a/dislocker_slash.py
+++ b/dislocker_slash.py
@@ -648,7 +648,7 @@ class DL():
     def pc_onetime_gen(self, **kwargs):
         if kwargs.get("max_count") == None:
             max_count = 1
-        elif kwargs.get("max_count").isdecimal():
+        elif isinstance(kwargs.get("max_count"), int):
             max_count = int(kwargs.get("max_count"))
         else:
             max_count = 1
@@ -700,7 +700,7 @@ class DL():
     def device_onetime_gen(self, **kwargs):
         if kwargs.get("max_count") == None:
             max_count = 1
-        elif kwargs.get("max_count").isdecimal():
+        elif isinstance(kwargs.get("max_count"), int):
             max_count = int(kwargs.get("max_count"))
         else:
             max_count = 1