diff --git a/register_types.cpp b/register_types.cpp
index 955e9a5..28b5d9d 100644
--- a/register_types.cpp
+++ b/register_types.cpp
@@ -22,6 +22,8 @@ SOFTWARE.
 
 #include "register_types.h"
 
+#include "core/object/class_db.h"
+
 #include "data/world_generator_prop_data.h"
 
 #include "main/building.h"
@@ -30,15 +32,17 @@ SOFTWARE.
 
 #include "world_generator.h"
 
-void register_world_generator_types() {
-	ClassDB::register_class<WorldGeneratorPropData>();
+void initialize_world_generator_module(ModuleInitializationLevel p_level) {
+	if (p_level == MODULE_INITIALIZATION_LEVEL_SCENE) {
+		GDREGISTER_CLASS(WorldGeneratorPropData);
 
-	ClassDB::register_class<Building>();
-	ClassDB::register_class<Biome>();
-	ClassDB::register_class<Planet>();
+		GDREGISTER_CLASS(Building);
+		GDREGISTER_CLASS(Biome);
+		GDREGISTER_CLASS(Planet);
 
-	ClassDB::register_class<WorldGenerator>();
+		GDREGISTER_CLASS(WorldGenerator);
+	}
 }
 
-void unregister_world_generator_types() {
+void uninitialize_world_generator_module(ModuleInitializationLevel p_level) {
 }
diff --git a/register_types.h b/register_types.h
index 8d753fb..9a8be17 100644
--- a/register_types.h
+++ b/register_types.h
@@ -20,5 +20,12 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
 */
 
-void register_world_generator_types();
-void unregister_world_generator_types();
+#ifndef WORLD_GENERATOR_REGISTER_TYPES_H
+#define WORLD_GENERATOR_REGISTER_TYPES_H
+
+#include "modules/register_module_types.h"
+
+void initialize_world_generator_module(ModuleInitializationLevel p_level);
+void uninitialize_world_generator_module(ModuleInitializationLevel p_level);
+
+#endif