Fixed the correct output of logistic regression