diff --git a/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/Variant.kt b/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/Variant.kt
index 3d9380d1..3b57d68c 100644
--- a/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/Variant.kt
+++ b/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/Variant.kt
@@ -56,12 +56,12 @@ class Variant(val initialProductFlavor: ProductFlavorConfig? = null,
 
 //            // The ordering of files is: 1) build type 2) product flavor 3) default
             buildType.let {
-                val dir = File(KFiles.joinDir("src", it.name, project.projectInfo.sourceDirectory))
+                val dir = File(KFiles.joinDir("src", it.name, suffix))
                 log(3, "Adding source for build type ${it.name}: ${dir.path}")
                 result.add(dir)
             }
             productFlavor.let {
-                val dir = File(KFiles.joinDir("src", it.name, project.projectInfo.sourceDirectory))
+                val dir = File(KFiles.joinDir("src", it.name, suffix))
                 log(3, "Adding source for product flavor ${it.name}: ${dir.path}")
                 result.add(dir)
             }