diff --git a/scripts/benchmarks/benchmark_gluonnlp.py b/scripts/benchmarks/benchmark_gluonnlp.py
index 7d4687e2b7..7d4c59b446 100644
--- a/scripts/benchmarks/benchmark_gluonnlp.py
+++ b/scripts/benchmarks/benchmark_gluonnlp.py
@@ -1,4 +1,6 @@
+import mxnet as mx
 from benchmark_utils import GluonNLPBackboneBenchmark
+mx.npx.set_np()
 
 
 MODELS = [