diff --git a/biome_terrarin_generator.cpp b/biome_terrarin_generator.cpp
index 1b897cb..b996786 100644
--- a/biome_terrarin_generator.cpp
+++ b/biome_terrarin_generator.cpp
@@ -22,9 +22,15 @@ SOFTWARE.
 
 #include "biome_terrarin_generator.h"
 
-#include "core/math/math_funcs.h"
+#include "core/version.h"
 
+#if VERSION_MAJOR > 3
+#include "core/config/engine.h"
+#else
 #include "core/engine.h"
+#endif
+
+#include "core/math/math_funcs.h"
 
 #include "../voxelman/world/default/voxel_chunk_default.h"
 #include "../voxelman/world/voxel_chunk.h"
diff --git a/biome_terrarin_generator.h b/biome_terrarin_generator.h
index 0b6f117..51b8614 100644
--- a/biome_terrarin_generator.h
+++ b/biome_terrarin_generator.h
@@ -23,7 +23,13 @@ SOFTWARE.
 #ifndef BIOME_TERRARIN_GENERATOR_H
 #define BIOME_TERRARIN_GENERATOR_H
 
+#include "core/version.h"
+
+#if VERSION_MAJOR > 3
+#include "core/object/reference.h"
+#else
 #include "core/reference.h"
+#endif
 
 class VoxelChunk;