diff --git a/Jenkinsfile-restricted b/Jenkinsfile-restricted
index 825d0bc4f..a9f1db9ae 100644
--- a/Jenkinsfile-restricted
+++ b/Jenkinsfile-restricted
@@ -48,22 +48,21 @@ pipeline {
             }
         }
         stage('Jenkins: Build doc') {
-            retry(3) {
-                agent {
-                    label 'linux && cpu && restricted'
-                }
-                steps {
-                    unstash name: 'srcs'
-                    script {
-                        def commit_id = "${GIT_COMMIT}"
-                        def branch_name = "${GIT_LOCAL_BRANCH}"
-                        echo 'Building doc...'
-                        dir ('jvm-packages') {
-                            sh "bash ./build_doc.sh ${commit_id}"
-                            archiveArtifacts artifacts: "${commit_id}.tar.bz2", allowEmptyArchive: true
-                            echo 'Deploying doc...'
-                            withAWS(credentials:'xgboost-doc-bucket') {
-                                s3Upload file: "${commit_id}.tar.bz2", bucket: 'xgboost-docs', acl: 'PublicRead', path: "${branch_name}.tar.bz2"
+            steps {
+                script {
+                    retry(3) {
+                        node('linux && cpu && restricted') {
+                            unstash name: 'srcs'
+                            def commit_id = "${GIT_COMMIT}"
+                            def branch_name = "${GIT_LOCAL_BRANCH}"
+                            echo 'Building doc...'
+                            dir ('jvm-packages') {
+                                sh "bash ./build_doc.sh ${commit_id}"
+                                archiveArtifacts artifacts: "${commit_id}.tar.bz2", allowEmptyArchive: true
+                                echo 'Deploying doc...'
+                                withAWS(credentials:'xgboost-doc-bucket') {
+                                    s3Upload file: "${commit_id}.tar.bz2", bucket: 'xgboost-docs', acl: 'PublicRead', path: "${branch_name}.tar.bz2"
+                                }
                             }
                         }
                     }