diff --git a/yarn/build.sh b/yarn/build.sh
index 12764357c..c86cff6a8 100755
--- a/yarn/build.sh
+++ b/yarn/build.sh
@@ -1,4 +1,4 @@
 #!/bin/bash
-CPATH=`${HADOOP_HOME}/bin/hadoop classpath`
+CPATH=`${HADOOP_PREFIX}/bin/hadoop classpath`
 javac -cp $CPATH -d bin src/org/apache/hadoop/yarn/rabit/*
 jar cf rabit-yarn.jar -C bin .