diff --git a/spark/spark-2.0/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala b/spark/spark-2.0/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala
index a68f88ff7..7d2cd8366 100644
--- a/spark/spark-2.0/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala
+++ b/spark/spark-2.0/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala
@@ -221,7 +221,7 @@ final class HivemallGroupedDataset(groupBy: RelationalGroupedDataset) {
   }
 
   /**
-   * @see hivemall.evaluation.FMeasureUDAF
+   * @see hivemall.evaluation.F1ScoreUDAF
    * @group evaluation
    */
   def f1score(predict: String, target: String): DataFrame = {
@@ -229,7 +229,7 @@ final class HivemallGroupedDataset(groupBy: RelationalGroupedDataset) {
     // checkType(predict, ArrayType(IntegerType))
     val udaf = HiveUDAFFunction(
         "f1score",
-        new HiveFunctionWrapper("hivemall.evaluation.FMeasureUDAF"),
+        new HiveFunctionWrapper("hivemall.evaluation.F1ScoreUDAF"),
         Seq(predict, target).map(df.col(_).expr),
         isUDAFBridgeRequired = true)
       .toAggregateExpression()
diff --git a/spark/spark-2.2/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala b/spark/spark-2.2/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala
index 75f4bfce3..a012efdb9 100644
--- a/spark/spark-2.2/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala
+++ b/spark/spark-2.2/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala
@@ -222,7 +222,7 @@ final class HivemallGroupedDataset(groupBy: RelationalGroupedDataset) {
   }
 
   /**
-   * @see hivemall.evaluation.FMeasureUDAF
+   * @see hivemall.evaluation.F1ScoreUDAF
    * @group evaluation
    */
   def f1score(predict: String, target: String): DataFrame = {
@@ -230,7 +230,7 @@ final class HivemallGroupedDataset(groupBy: RelationalGroupedDataset) {
     // checkType(predict, ArrayType(IntegerType))
     val udaf = HiveUDAFFunction(
         "f1score",
-        new HiveFunctionWrapper("hivemall.evaluation.FMeasureUDAF"),
+        new HiveFunctionWrapper("hivemall.evaluation.F1ScoreUDAF"),
         Seq(predict, target).map(df.col(_).expr),
         isUDAFBridgeRequired = true)
       .toAggregateExpression()