diff --git a/src/main.cpp b/src/main.cpp
index 61558cf..80346ff 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -7,7 +7,10 @@
 #include <assimp/Importer.hpp>
 #include <assimp/scene.h>
 #include <assimp/postprocess.h>
+
+#ifdef DEBUG
 #include <imgui.h>
+#endif
 
 #include "renderer.h"
 #include "platform.h"
@@ -490,6 +493,10 @@ int main(int argc, char* argv[]) {
         delete mesh;
     }
     
+    for(auto light : world->lights) {
+        delete light;
+    }
+    
     delete world;
 
     renderer->destroyRenderTarget(target);