diff --git a/example/autoencoder/model.py b/example/autoencoder/model.py
index c1b72216ef98..9b6185c9fd18 100644
--- a/example/autoencoder/model.py
+++ b/example/autoencoder/model.py
@@ -22,7 +22,7 @@
 import numpy as np
 try:
     import cPickle as pickle
-except ModuleNotFoundError:
+except ImportError:
     import pickle