1 files changed, 2 insertions, 0 deletions
diff --git a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java
index e41fbb8..717715e 100644
--- a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java
+++ b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java
@@ -15,6 +15,7 @@ public class SqlUnitModel {
     private Map<SequenceMirror, Element> sequenceElements = new TreeMap<>();
     private String packageName;
     private String name;
+    public TypeRef sessionType;
 
     public String getPackageName() {
         if (packageName == null) {
@@ -34,6 +35,7 @@ public class SqlUnitModel {
 
     public void setName(String name) {
         this.name = name;
+        sessionType = new TypeRef(packageName + "." + name + "SqlSession");
     }
 
     // -----------------------------------------------------------------------