DRILL-148 - Updated to adapt to changes in underlying Mahout code.