[SPARK-1161] Fixing doc tests