diff --git a/jenkins/pytorch/build_nimbix.sh b/jenkins/pytorch/build_nimbix.sh
index 1933565fd..08e3a6a41 100644
--- a/jenkins/pytorch/build_nimbix.sh
+++ b/jenkins/pytorch/build_nimbix.sh
@@ -217,6 +217,7 @@ if [ ! -z "$jenkins_nightly" ]; then
     git clone https://github.com/ezyang/onnx-pytorch.git --recurse-submodules --quiet
     (cd onnx-pytorch/onnx && python setup.py install)
     (cd onnx-pytorch/onnx-caffe2 && python setup.py install)
+    (cd onnx-pytorch && python setup.py install)
     python onnx-pytorch/test/test_models.py
     python onnx-pytorch/test/test_caffe2.py
 fi