fix for spark-test and mllib tests