Fixing the neural net model.