diff --git a/compiler/src/main/java/se/emilsjolander/intentbuilder/Processor.java b/compiler/src/main/java/se/emilsjolander/intentbuilder/Processor.java index fb51182..6d962f3 100644 --- a/compiler/src/main/java/se/emilsjolander/intentbuilder/Processor.java +++ b/compiler/src/main/java/se/emilsjolander/intentbuilder/Processor.java @@ -153,7 +153,9 @@ private TypeSpec getBuilderSpec(Element annotatedElement) { .addStatement("$T extras = intent.getExtras()", Bundle.class); for (Element e : all) { String paramName = e.getSimpleName().toString(); + injectMethod.beginControlFlow("if (extras.containsKey($S))", paramName); injectMethod.addStatement("activity.$N = ($T) extras.get($S)", paramName, e.asType(), paramName); + injectMethod.endControlFlow(); } builder.addMethod(injectMethod.build());