diff --git a/jte-kotlin/src/main/java/gg/jte/compiler/kotlin/KotlinCodeGenerator.java b/jte-kotlin/src/main/java/gg/jte/compiler/kotlin/KotlinCodeGenerator.java
index 42b0c04f..74d381ae 100644
--- a/jte-kotlin/src/main/java/gg/jte/compiler/kotlin/KotlinCodeGenerator.java
+++ b/jte-kotlin/src/main/java/gg/jte/compiler/kotlin/KotlinCodeGenerator.java
@@ -48,6 +48,7 @@ public KotlinCodeGenerator(TemplateCompiler compiler, TemplateConfig config, Con
     @Override
     public void onImport(String importClass) {
         writePackageIfRequired();
+        imports.add(importClass);
         kotlinCode.append("import ").append(importClass).append("\n");
     }