Fix warnings
closes #13
diff --git a/src/main/scala/Engine.scala b/src/main/scala/Engine.scala
index 16d47ac..abe8778 100644
--- a/src/main/scala/Engine.scala
+++ b/src/main/scala/Engine.scala
@@ -1,6 +1,6 @@
package org.example.textclassification
-import org.apache.predictionio.controller.IEngineFactory
+import org.apache.predictionio.controller.EngineFactory
import org.apache.predictionio.controller.Engine
/** Define Query class which serves as a wrapper for
@@ -22,7 +22,7 @@
case class ActualResult(category: String) extends Serializable
/** Define Engine */
-object TextClassificationEngine extends IEngineFactory {
+object TextClassificationEngine extends EngineFactory {
def apply() = {
new Engine(
classOf[DataSource],