diff --git a/delivery-sdk-generators/src/main/java/kontent/ai/delivery/generators/CodeGenerator.java b/delivery-sdk-generators/src/main/java/kontent/ai/delivery/generators/CodeGenerator.java
index 84c4fde9..f0d1eaae 100644
--- a/delivery-sdk-generators/src/main/java/kontent/ai/delivery/generators/CodeGenerator.java
+++ b/delivery-sdk-generators/src/main/java/kontent/ai/delivery/generators/CodeGenerator.java
@@ -140,7 +140,7 @@ public JavaFile generateSource(ContentType type) {
             if (typeName != null) {
                 //Add the field
                 String fieldName = CaseFormat.LOWER_UNDERSCORE.to(CaseFormat.LOWER_CAMEL, element.getKey());
-                Class annoClass = element.getValue().getType() == "modular_content" ?
+                Class annoClass = element.getValue().getType().equals("modular_content") ?
                         ContentItemMapping.class : ElementMapping.class;
                 fieldSpecs.add(
                         FieldSpec.builder(typeName, fieldName)