1 files changed, 5 insertions, 3 deletions
diff --git a/container-compiler-plugin/src/main/java/io/trygvis/persistence/GeneratorSupport.java b/container-compiler-plugin/src/main/java/io/trygvis/persistence/GeneratorSupport.java
index d3293c3..379f6f6 100644
--- a/container-compiler-plugin/src/main/java/io/trygvis/persistence/GeneratorSupport.java
+++ b/container-compiler-plugin/src/main/java/io/trygvis/persistence/GeneratorSupport.java
@@ -2,6 +2,7 @@ package io.trygvis.persistence;
 
 import javax.lang.model.element.Element;
 import javax.lang.model.type.TypeMirror;
+import java.util.Date;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -11,9 +12,10 @@ public class GeneratorSupport {
     private final Map<String, TypeHandler> typeHandlers = new HashMap<>();
 
     {
-        typeHandlers.put("java.lang.Integer", new TypeHandler.IntTypeHandler());
-        typeHandlers.put("java.lang.Long", new TypeHandler.LongTypeHandler());
-        typeHandlers.put("java.util.Date", new TypeHandler.DateTypeHandler());
+        typeHandlers.put(Integer.class.getName(), new TypeHandler.IntTypeHandler());
+        typeHandlers.put(Long.class.getName(), new TypeHandler.LongTypeHandler());
+        typeHandlers.put(String.class.getName(), new TypeHandler.StringTypeHandler());
+        typeHandlers.put(Date.class.getName(), new TypeHandler.DateTypeHandler());
 
         primitiveTypeHandlers.putAll(typeHandlers);
     }