Fixed bug in test set accuracy computation