diff --git a/sklearn_porter/estimator/classifier/DecisionTreeClassifier/templates/ruby/embedded.method.txt b/sklearn_porter/estimator/classifier/DecisionTreeClassifier/templates/ruby/embedded.method.txt
index d751203b..d5781990 100644
--- a/sklearn_porter/estimator/classifier/DecisionTreeClassifier/templates/ruby/embedded.method.txt
+++ b/sklearn_porter/estimator/classifier/DecisionTreeClassifier/templates/ruby/embedded.method.txt
@@ -1,7 +1,7 @@
-def self.{method_name} (atts)
+def self.{method_name} (features)
 	classes = Array.new({n_classes}, 0)
 	{branches}
 
 	pos = classes.each_with_index.select {{|e, i| e==classes.max}}.map &:last
 	return pos.min
-end
\ No newline at end of file
+end