diff --git a/tf_keras/preprocessing/sequence.py b/tf_keras/preprocessing/sequence.py
index 291c7638e..cabcd143c 100644
--- a/tf_keras/preprocessing/sequence.py
+++ b/tf_keras/preprocessing/sequence.py
@@ -375,7 +375,7 @@ def skipgrams(
 
     if shuffle:
         if seed is None:
-            seed = random.randint(0, 10e6)
+            seed = random.randint(0, int(10e6))
         random.seed(seed)
         random.shuffle(couples)
         random.seed(seed)
diff --git a/tf_keras/utils/text_dataset_test.py b/tf_keras/utils/text_dataset_test.py
index 50349618b..df4419899 100644
--- a/tf_keras/utils/text_dataset_test.py
+++ b/tf_keras/utils/text_dataset_test.py
@@ -33,7 +33,7 @@ def _prepare_directory(
     ):
         # Get a unique temp directory
         temp_dir = os.path.join(
-            self.get_temp_dir(), str(random.randint(0, 1e6))
+            self.get_temp_dir(), str(random.randint(0, int(1e6)))
         )
         os.mkdir(temp_dir)
         self.addCleanup(shutil.rmtree, temp_dir)