diff --git a/docs/_static/code/zh/tfhub/stylization.py b/docs/_static/code/zh/tfhub/stylization.py
index c365d93d..833ac941 100644
--- a/docs/_static/code/zh/tfhub/stylization.py
+++ b/docs/_static/code/zh/tfhub/stylization.py
@@ -43,6 +43,7 @@ def load_image_local(image_path, image_size=(512, 512), preserve_aspect_ratio=Tr
     return img
 
 def show_image(image, title, save=False):
+    fig = plt.figure()
     plt.imshow(image, aspect='equal')
     plt.axis('off')
     if save:
diff --git a/docs/_static/code/zh/tools/save_and_restore/mnist.py b/docs/_static/code/zh/tools/save_and_restore/mnist.py
index 0e663b27..cbb6e8c2 100644
--- a/docs/_static/code/zh/tools/save_and_restore/mnist.py
+++ b/docs/_static/code/zh/tools/save_and_restore/mnist.py
@@ -6,9 +6,9 @@
 
 parser = argparse.ArgumentParser(description='Process some integers.')
 parser.add_argument('--mode', default='train', help='train or test')
-parser.add_argument('--num_epochs', default=1)
-parser.add_argument('--batch_size', default=50)
-parser.add_argument('--learning_rate', default=0.001)
+parser.add_argument('--num_epochs', default=1, type=int)
+parser.add_argument('--batch_size', default=50, type=int)
+parser.add_argument('--learning_rate', default=0.001, type=float)
 args = parser.parse_args()
 data_loader = MNISTLoader()
 
diff --git a/docs/_static/code/zh/tools/save_and_restore/mnist_manager.py b/docs/_static/code/zh/tools/save_and_restore/mnist_manager.py
index ec2848a5..27876aa7 100644
--- a/docs/_static/code/zh/tools/save_and_restore/mnist_manager.py
+++ b/docs/_static/code/zh/tools/save_and_restore/mnist_manager.py
@@ -6,9 +6,9 @@
 
 parser = argparse.ArgumentParser(description='Process some integers.')
 parser.add_argument('--mode', default='train', help='train or test')
-parser.add_argument('--num_epochs', default=1)
-parser.add_argument('--batch_size', default=50)
-parser.add_argument('--learning_rate', default=0.001)
+parser.add_argument('--num_epochs', default=1, type=int)
+parser.add_argument('--batch_size', default=50, type=int)
+parser.add_argument('--learning_rate', default=0.001, type=float)
 args = parser.parse_args()
 data_loader = MNISTLoader()