Naprawienie działania NaiveBayes