diff --git a/Jenkinsfile b/Jenkinsfile
index 7a2afb3fdf875..92d9e535963d8 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -45,7 +45,7 @@
 //
 
 ci_lint = "tvmai/ci-lint:v0.51"
-ci_gpu = "tvmai/ci-gpu:v0.56"
+ci_gpu = "tvmai/ci-gpu:v0.60"
 ci_cpu = "tvmai/ci-cpu:v0.55"
 ci_i386 = "tvmai/ci-i386:v0.52"
 
diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
index fc05e7a011a99..f9e8aac4e389a 100644
--- a/tests/python/frontend/onnx/test_forward.py
+++ b/tests/python/frontend/onnx/test_forward.py
@@ -16,6 +16,8 @@
 # under the License.
 import numpy as np
 import math
+import onnx
+from onnx import helper, TensorProto, mapping
 import torch
 import torchvision
 import topi
@@ -24,8 +26,6 @@
 from tvm import relay
 from tvm.contrib import graph_runtime
 from tvm.relay.testing.config import ctx_list
-import onnx
-from onnx import helper, TensorProto, mapping
 import scipy