diff --git a/buildSrc/src/main/java/org/ultramine/gradle/task/SpeicialClassTransformTask.java b/buildSrc/src/main/java/org/ultramine/gradle/task/SpeicialClassTransformTask.java
index f6ab49b..0a5a70f 100644
--- a/buildSrc/src/main/java/org/ultramine/gradle/task/SpeicialClassTransformTask.java
+++ b/buildSrc/src/main/java/org/ultramine/gradle/task/SpeicialClassTransformTask.java
@@ -4,6 +4,7 @@
 import org.apache.commons.io.FileUtils;
 import org.gradle.api.DefaultTask;
 import org.gradle.api.tasks.Input;
+import org.gradle.api.tasks.InputDirectory;
 import org.gradle.api.tasks.OutputDirectory;
 import org.gradle.api.tasks.ParallelizableTask;
 import org.gradle.api.tasks.TaskAction;
@@ -32,6 +33,7 @@
 @ParallelizableTask
 public class SpeicialClassTransformTask extends DefaultTask
 {
+	@InputDirectory
 	private File inputDir;
 	@OutputDirectory
 	private File outputDir = new File(getProject().getBuildDir(), getName());
@@ -116,7 +118,7 @@
 		if(file.isDirectory())
 			return;
 		String path = getRelPath(file);
-		processClass(path, transformerMap.get(file));
+		processClass(path, transformerMap.get(path));
 	}
 
 	private void processClass(String path, List<ISpecialTransformer> transfs)
@@ -211,5 +213,25 @@
 		{
 			replaceMap.put(search, replacement);
 		}
+
+		@Override
+		public boolean equals(Object o)
+		{
+			if(this == o) return true;
+			if(o == null || getClass() != o.getClass()) return false;
+
+			ReplaceStringTransformer that = (ReplaceStringTransformer) o;
+
+			if(path != null ? !path.equals(that.path) : that.path != null) return false;
+			return replaceMap != null ? replaceMap.equals(that.replaceMap) : that.replaceMap == null;
+		}
+
+		@Override
+		public int hashCode()
+		{
+			int result = path != null ? path.hashCode() : 0;
+			result = 31 * result + (replaceMap != null ? replaceMap.hashCode() : 0);
+			return result;
+		}
 	}
 }