diff --git a/data/world_generator_prop_data.h b/data/world_generator_prop_data.h
index 8711799..45bb423 100644
--- a/data/world_generator_prop_data.h
+++ b/data/world_generator_prop_data.h
@@ -23,7 +23,13 @@ SOFTWARE.
 #ifndef WORLD_GENERATOR_PROP_DATA_H
 #define WORLD_GENERATOR_PROP_DATA_H
 
+#include "core/version.h"
+
+#if VERSION_MAJOR > 3
+#include "core/io/resource.h"
+#else
 #include "core/resource.h"
+#endif
 
 class PackedScene;
 
diff --git a/main/biome.h b/main/biome.h
index 61a6c5a..a375d0a 100644
--- a/main/biome.h
+++ b/main/biome.h
@@ -23,8 +23,15 @@ SOFTWARE.
 #ifndef BIOME_H
 #define BIOME_H
 
+#include "core/version.h"
+
+#if VERSION_MAJOR > 3
+#include "core/io/resource.h"
+#include "core/object/reference.h"
+#else
 #include "core/reference.h"
 #include "core/resource.h"
+#endif
 
 #include "dungeon.h"
 
diff --git a/main/dungeon.h b/main/dungeon.h
index b0cbda6..bbd549f 100644
--- a/main/dungeon.h
+++ b/main/dungeon.h
@@ -23,8 +23,15 @@ SOFTWARE.
 #ifndef DUNGEON_H
 #define DUNGEON_H
 
+#include "core/version.h"
+
+#if VERSION_MAJOR > 3
+#include "core/io/resource.h"
+#include "core/object/reference.h"
+#else
 #include "core/reference.h"
 #include "core/resource.h"
+#endif
 
 #include "dungeon_corridor.h"
 #include "dungeon_room.h"
diff --git a/main/dungeon_room.h b/main/dungeon_room.h
index 232f60e..a353395 100644
--- a/main/dungeon_room.h
+++ b/main/dungeon_room.h
@@ -23,9 +23,17 @@ SOFTWARE.
 #ifndef DUNGEON_ROOM_H
 #define DUNGEON_ROOM_H
 
+#include "core/version.h"
+
+#if VERSION_MAJOR > 3
+#include "core/io/resource.h"
+#include "core/object/reference.h"
+#include "core/templates/vector.h"
+#else
 #include "core/reference.h"
 #include "core/resource.h"
 #include "core/vector.h"
+#endif
 
 #include "../data/world_generator_prop_data.h"
 
diff --git a/main/planet.h b/main/planet.h
index 94611d1..1cc86d5 100644
--- a/main/planet.h
+++ b/main/planet.h
@@ -23,9 +23,17 @@ SOFTWARE.
 #ifndef PLANET_H
 #define PLANET_H
 
+#include "core/version.h"
+
+#if VERSION_MAJOR > 3
+#include "core/io/image.h"
+#include "core/io/resource.h"
+#include "core/object/reference.h"
+#else
 #include "core/image.h"
 #include "core/reference.h"
 #include "core/resource.h"
+#endif
 
 #include "biome.h"
 #include "dungeon.h"
diff --git a/world_generator.h b/world_generator.h
index f2b40d2..cd4e4c9 100644
--- a/world_generator.h
+++ b/world_generator.h
@@ -23,8 +23,15 @@ SOFTWARE.
 #ifndef WORLD_GENERATOR_H
 #define WORLD_GENERATOR_H
 
+#include "core/version.h"
+
+#if VERSION_MAJOR > 3
+#include "core/io/resource.h"
+#include "core/templates/vector.h"
+#else
 #include "core/resource.h"
 #include "core/vector.h"
+#endif
 
 #include "main/planet.h"