1 files changed, 1 insertions, 1 deletions
diff --git a/feature.rb b/feature.rb
index 3b39fa9..15ae4c9 100644
--- a/feature.rb
+++ b/feature.rb
@@ -7,7 +7,7 @@ class FeatureTest < Test::Unit::TestCase
 
   def setup
     @features = [
-      "http://localhost/dataset/1/feature/Hamster%20Carcinogenicity",
+      @@classification_training_dataset.features.keys.first,
       "http://apps.ideaconsult.net:8080/ambit2/feature/35796"
     ]
   end