1 files changed, 11 insertions, 2 deletions
diff --git a/lazar.rb b/lazar.rb
index 5de3790..f4915a7 100644
--- a/lazar.rb
+++ b/lazar.rb
@@ -61,8 +61,7 @@ post '/lazar/?' do
 
 		lazar = OpenTox::Model::Lazar.new
     lazar.min_sim = params[:min_sim].to_f if params[:min_sim]
-    lazar.nr_hits = true if params[:nr_hits] == "true"
-
+    
     if prediction_feature.feature_type == "classification"
       @training_classes = training_activities.accept_values(prediction_feature.uri).sort
       @training_classes.each_with_index { |c,i|
@@ -70,8 +69,18 @@ post '/lazar/?' do
         params[:value_map] = lazar.value_map
       }
     elsif  prediction_feature.feature_type == "regression"
+      lazar.nr_hits = true
       lazar.prediction_algorithm = "Neighbors.local_svm_regression" 
     end
+
+    if params[:nr_hits] == "false" # if nr_hits is set optional to true/false it will return as String (but should be True/FalseClass)
+      lazar.nr_hits = false
+      #params[:nr_hits] = false
+    elsif params[:nr_hits] == "true"
+      lazar.nr_hits = true
+    end
+    params[:nr_hits] = "true" if lazar.nr_hits
+
     task.progress 10
 
 		if params[:feature_dataset_uri]