fixing for sklearn 0.14.x