diff --git a/tensorflow_graphics/projects/gan/keras_layers.py b/tensorflow_graphics/projects/gan/keras_layers.py
index 4580dbdb1..c74fedd6c 100644
--- a/tensorflow_graphics/projects/gan/keras_layers.py
+++ b/tensorflow_graphics/projects/gan/keras_layers.py
@@ -393,14 +393,14 @@ def call(
   def get_config(self) -> Dict[str, Any]:
     """Returns the config of the layer."""
     config = {
-        'kernel_size':
-            self._kernel_size,
-        'filters':
-            self._filters,
-        'kernel_initializer':
-            tf.keras.utils.serialize_keras_object(self._kernel_initializer),
-        'bias_initializer':
-            tf.keras.utils.serialize_keras_object(self._bias_initializer),
+        'kernel_size': self._kernel_size,
+        'filters': self._filters,
+        'kernel_initializer': tf.keras.utils.legacy.serialize_keras_object(
+            self._kernel_initializer
+        ),
+        'bias_initializer': tf.keras.utils.legacy.serialize_keras_object(
+            self._bias_initializer
+        ),
     }
     base_config = super().get_config()
     base_config.update(config)