diff --git a/editor/build.gradle b/editor/build.gradle
index d83189af..228d4ea8 100644
--- a/editor/build.gradle
+++ b/editor/build.gradle
@@ -36,19 +36,24 @@ dependencies {
     implementation project(":proto")
 
     implementation "org.jetbrains.kotlin:kotlin-stdlib"
+
+    // GUI
     implementation "no.tornado:tornadofx:${tornadoFxVersion}"
     implementation platform("org.kordamp.ikonli:ikonli-bom:${ikonliVersion}")
     implementation 'org.kordamp.ikonli:ikonli-javafx'
     implementation 'org.kordamp.ikonli:ikonli-fontawesome-pack'
-    implementation "org.fxmisc.richtext:richtextfx:${richtextfxVersion}"
-    implementation "org.codehaus.janino:janino:${janinoVersion}"
-    implementation "org.codehaus.janino:commons-compiler:${janinoVersion}"
 
     // Spring
     implementation 'org.springframework.boot:spring-boot-starter'
     implementation 'org.springframework.boot:spring-boot-starter-freemarker'
     implementation "org.slf4j:jul-to-slf4j:${slf4jVersion}"
 
+    // Code
+    implementation "org.fxmisc.richtext:richtextfx:${richtextfxVersion}"
+    implementation "org.codehaus.janino:janino:${janinoVersion}"
+    implementation "org.codehaus.janino:commons-compiler:${janinoVersion}"
+    implementation "com.squareup:javapoet:${javaPoetVersion}"
+
     // Database
     implementation "com.h2database:h2:${h2Version}"
     implementation "com.zaxxer:HikariCP:${hikariVersion}"
diff --git a/gradle.properties b/gradle.properties
index 48b644e3..4c84abc5 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -14,4 +14,5 @@ richtextfxVersion=0.10.5
 janinoVersion=3.1.3
 apacheCommonsVersion=3.6.1
 h2Version=1.4.200
-hikariVersion=4.0.3
\ No newline at end of file
+hikariVersion=4.0.3
+javaPoetVersion=1.13.0
\ No newline at end of file