diff --git a/mlpp/gan/gan.h b/mlpp/gan/gan.h
index eb7cde8..5717d7c 100644
--- a/mlpp/gan/gan.h
+++ b/mlpp/gan/gan.h
@@ -10,6 +10,8 @@
 
 #include "core/math/math_defs.h"
 
+#include "core/object/reference.h"
+
 #include "../hidden_layer/hidden_layer.h"
 #include "../output_layer/output_layer.h"
 
@@ -20,7 +22,9 @@
 #include <tuple>
 #include <vector>
 
-class MLPPGAN {
+class MLPPGAN : public Reference {
+	GDCLASS(MLPPGAN, Reference);
+
 public:
 	/*
 	Ref<MLPPMatrix> get_input_set();
diff --git a/register_types.cpp b/register_types.cpp
index cbaacc0..4cff229 100644
--- a/register_types.cpp
+++ b/register_types.cpp
@@ -59,6 +59,7 @@ SOFTWARE.
 #include "mlpp/log_reg/log_reg.h"
 #include "mlpp/lin_reg/lin_reg.h"
 #include "mlpp/gaussian_nb/gaussian_nb.h"
+#include "mlpp/gan/gan.h"
 
 #include "test/mlpp_tests.h"
 
@@ -100,6 +101,7 @@ void register_pmlpp_types(ModuleRegistrationLevel p_level) {
 		ClassDB::register_class<MLPPLogReg>();
 		ClassDB::register_class<MLPPLinReg>();
 		ClassDB::register_class<MLPPGaussianNB>();
+		ClassDB::register_class<MLPPGAN>();
 
 		ClassDB::register_class<MLPPDataESimple>();
 		ClassDB::register_class<MLPPDataSimple>();