diff --git a/whetstone-compiler/src/main/java/com/deliveryhero/whetstone/compiler/handlers/AppComponentHandler.kt b/whetstone-compiler/src/main/java/com/deliveryhero/whetstone/compiler/handlers/AppComponentHandler.kt
index 8ac13af..812b57c 100644
--- a/whetstone-compiler/src/main/java/com/deliveryhero/whetstone/compiler/handlers/AppComponentHandler.kt
+++ b/whetstone-compiler/src/main/java/com/deliveryhero/whetstone/compiler/handlers/AppComponentHandler.kt
@@ -67,12 +67,11 @@ internal class AppComponentHandler : CodegenHandler {
             .addFunction(
                 FunSpec.builder("create")
                     .addParameter(appParam)
-                    .returns(generatedComponentCn)
+                    .returns(applicationComponentCn)
                     .addModifiers(KModifier.PUBLIC, KModifier.OVERRIDE)
                     .addCode(CodeBlock.of(
-                        "return DaggerGeneratedApplicationComponent.factory().create(%L) as %T",
-                        appParam.name,
-                        generatedComponentCn,
+                        "return DaggerGeneratedApplicationComponent.factory().create(%L)",
+                        appParam.name
                     ))
                     .build())
             .build()