Fixing naive bayes integration test