diff --git a/template/plugin/pdk_types.py.ejs b/template/plugin/pdk_types.py.ejs
index be16f80..cdcd11e 100644
--- a/template/plugin/pdk_types.py.ejs
+++ b/template/plugin/pdk_types.py.ejs
@@ -21,7 +21,20 @@ class <%- capitalize(schema.name) %>(extism.Json):
 <% if (p.description) { -%>
   # <%- formatCommentBlock(p.description, "# ") %>
 <% } -%>
+<% if (!p.nullable) {%>
   <%- p.name %>: <%- toPythonType(p) %>
+<% } %>
+<% }) %>
+
+<% schema.properties.forEach(p => { -%>
+<% if (p.description) { -%>
+  # <%- formatCommentBlock(p.description, "# ") %>
+<% } -%>
+<% if (p.nullable) {%>
+  <%- p.name %>: <%- toPythonType(p) %> = None
+<% } %>
 <% }) %>
+
 <% } %>
 <% }); %>
+