diff --git a/Jenkinsfile b/Jenkinsfile
index 60ee14249d28..1d711ddefc6b 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -259,7 +259,7 @@ stage('Integration Test') {
     }
   },
   'docs: GPU': {
-    node('GPU') {
+    node('TensorCore') {
       ws(per_exec_ws("tvm/docs-python-gpu")) {
         init_git()
         unpack_lib('gpu', tvm_multilib)
diff --git a/tutorials/frontend/deploy_prequantized_tflite.py b/tutorials/frontend/deploy_prequantized_tflite.py
index f6c4544a7fa3..3cdd4233b0cd 100644
--- a/tutorials/frontend/deploy_prequantized_tflite.py
+++ b/tutorials/frontend/deploy_prequantized_tflite.py
@@ -114,8 +114,13 @@ def get_real_image(im_height, im_width):
 tflite_model_file = os.path.join(model_dir, "mobilenet_v2_1.0_224_quant.tflite")
 tflite_model_buf = open(tflite_model_file, "rb").read()
 
-tflite_model = tflite.Model.GetRootAsModel(tflite_model_buf, 0)
-
+# Get TFLite model from buffer
+try:
+    import tflite
+    tflite_model = tflite.Model.GetRootAsModel(tflite_model_buf, 0)
+except AttributeError:
+    import tflite.Model
+    tflite_model = tflite.Model.Model.GetRootAsModel(tflite_model_buf, 0)
 
 ###############################################################################
 # Lets run TFLite pre-quantized model inference and get the TFLite prediction.