minor test fixes

1 files changed, 1 insertions, 1 deletions

diff --git a/test/classification-model.rb b/test/classification-model.rb
index 8cbd4bb..2032bf8 100644
--- a/test/classification-model.rb
+++ b/test/classification-model.rb
@@ -89,7 +89,7 @@ class ClassificationModelTest < MiniTest::Test
     assert_equal ["non-carcinogenic"], result.values(result.compounds[5], prediction_feature)
     assert_nil result.predictions[result.compounds.first][:value]
     assert_equal "carcinogenic", result.predictions[result.compounds[1]][:value]
-    assert_equal 0.27, result.predictions[result.compounds[1]][:probabilities]["no"].round(2)
+    assert_equal 0.27, result.predictions[result.compounds[1]][:probabilities]["non-carcinogenic"].round(2)
   end
 
   def test_carcinogenicity_rf_classification