Fix ModelKNN