diff --git a/navigation/src/main/java/me/foolishchow/android/plugin/navigation/NavigationPlugin.kt b/navigation/src/main/java/me/foolishchow/android/plugin/navigation/NavigationPlugin.kt
index 7aeb3be..e3efc81 100644
--- a/navigation/src/main/java/me/foolishchow/android/plugin/navigation/NavigationPlugin.kt
+++ b/navigation/src/main/java/me/foolishchow/android/plugin/navigation/NavigationPlugin.kt
@@ -2,7 +2,10 @@ package me.foolishchow.android.plugin.navigation
 
 import com.android.build.gradle.api.ApplicationVariant
 import com.android.build.gradle.internal.dsl.BaseAppModuleExtension
-import me.foolishchow.android.navigationprocessor.extensions.*
+import me.foolishchow.android.plugin.navigation.extensions.NavigationTaskName
+import me.foolishchow.android.plugin.navigation.extensions.navJavaDir
+import me.foolishchow.android.plugin.navigation.extensions.navResDir
+import me.foolishchow.android.plugin.navigation.extensions.navRuleFile
 import org.gradle.api.Plugin
 import org.gradle.api.Project
 import org.gradle.kotlin.dsl.closureOf
diff --git a/navigation/src/main/java/me/foolishchow/android/plugin/navigation/NavigationTask.kt b/navigation/src/main/java/me/foolishchow/android/plugin/navigation/NavigationTask.kt
index dfb5483..c73df1b 100644
--- a/navigation/src/main/java/me/foolishchow/android/plugin/navigation/NavigationTask.kt
+++ b/navigation/src/main/java/me/foolishchow/android/plugin/navigation/NavigationTask.kt
@@ -7,12 +7,8 @@ import com.squareup.javapoet.MethodSpec
 import com.squareup.javapoet.TypeSpec
 import groovy.util.Node
 import groovy.util.XmlParser
-import me.foolishchow.android.navigationprocessor.extensions.resId
-import me.foolishchow.android.navigationprocessor.extensions.resourceSymbol
-import me.foolishchow.android.navigationprocessor.extensions.snake2camel
-import org.apache.tools.ant.taskdefs.condition.IsLastModified
+import me.foolishchow.android.plugin.navigation.extensions.*
 import org.gradle.api.DefaultTask
-import org.gradle.api.artifacts.transform.InputArtifact
 import org.gradle.api.file.ConfigurableFileCollection
 import org.gradle.api.tasks.*
 import java.io.BufferedWriter
diff --git a/navigation/src/main/java/me/foolishchow/android/plugin/navigation/RuleEditor.kt b/navigation/src/main/java/me/foolishchow/android/plugin/navigation/RuleEditor.kt
index 9389a37..247307f 100644
--- a/navigation/src/main/java/me/foolishchow/android/plugin/navigation/RuleEditor.kt
+++ b/navigation/src/main/java/me/foolishchow/android/plugin/navigation/RuleEditor.kt
@@ -1,8 +1,7 @@
 package me.foolishchow.android.plugin.navigation
 
 import com.android.build.gradle.api.ApplicationVariant
-import me.foolishchow.android.navigationprocessor.extensions.AaptRules
-import me.foolishchow.android.navigationprocessor.extensions.navRuleFile
+import me.foolishchow.android.plugin.navigation.extensions.*
 import org.gradle.api.Project
 import java.io.BufferedReader
 import java.io.BufferedWriter