fixing log loss