[SPARK-1817] Correcting build error introduced in previous update