diff --git a/modules/wfc/register_types.cpp b/modules/wfc/register_types.cpp
index e84abb5ab..2a56378d8 100644
--- a/modules/wfc/register_types.cpp
+++ b/modules/wfc/register_types.cpp
@@ -1,7 +1,14 @@
 
 #include "register_types.h"
 
+#include "overlapping_wave_form_collapse.h"
+#include "tiling_wave_form_collapse.h"
+#include "wave_form_collapse.h"
+
 void register_wfc_types() {
+	ClassDB::register_class<WaveFormCollapse>();
+	ClassDB::register_class<OverlappingWaveFormCollapse>();
+	ClassDB::register_class<TilingWaveFormCollapse>();
 }
 
 void unregister_wfc_types() {
diff --git a/modules/wfc/tiling_wave_form_collapse.h b/modules/wfc/tiling_wave_form_collapse.h
index 5b4e4a986..8e591b57a 100644
--- a/modules/wfc/tiling_wave_form_collapse.h
+++ b/modules/wfc/tiling_wave_form_collapse.h
@@ -41,7 +41,7 @@ struct Tile {
 	Tile(const Array2D<uint32_t> &p_data, Symmetry p_symmetry, double p_weight);
 };
 
-class TilingWaveFormCollapse : WaveFormCollapse {
+class TilingWaveFormCollapse : public WaveFormCollapse {
 	GDCLASS(TilingWaveFormCollapse, WaveFormCollapse);
 
 public: