diff --git a/build.gradle b/build.gradle
index baf2819..90a7289 100644
--- a/build.gradle
+++ b/build.gradle
@@ -4,7 +4,7 @@ plugins {
 }
 
 group 'com.madrapps'
-version '1.0'
+version '1.1'
 
 repositories {
     mavenCentral()
diff --git a/src/main/kotlin/com/madrapps/daggerprocessor/DaggerProcessor.kt b/src/main/kotlin/com/madrapps/daggerprocessor/DaggerProcessor.kt
index c4728e2..131b62f 100644
--- a/src/main/kotlin/com/madrapps/daggerprocessor/DaggerProcessor.kt
+++ b/src/main/kotlin/com/madrapps/daggerprocessor/DaggerProcessor.kt
@@ -29,7 +29,7 @@ class DaggerProcessor : AbstractProcessor() {
         roundEnv.getElementsAnnotatedWith(GenerateTest::class.java).forEach { element ->
             if (element is Symbol.ClassSymbol) {
                 val ann: GenerateTest = element.getAnnotation(GenerateTest::class.java)
-                val packageName = element.owner.qualifiedName.toString()
+                val packageName = element.owner.qualifiedName.toString() + "." + ann.directory
                 val className = element.simpleName.toString()
                 val injectFile = File(testDateFile, ann.directory)
 
@@ -41,6 +41,8 @@ class DaggerProcessor : AbstractProcessor() {
                             """
 package $packageName
 
+import com.madrapps.dagger.$className
+
 class $testClassName : $className() {
 ${getTestMethods(folder)}
 }