fixed a bug, where the new model generated 1 extra feature