diff --git a/tensorflow_graphics/datasets/modelnet40/modelnet40.py b/tensorflow_graphics/datasets/modelnet40/modelnet40.py
index bd958b6a1..9dee09b24 100644
--- a/tensorflow_graphics/datasets/modelnet40/modelnet40.py
+++ b/tensorflow_graphics/datasets/modelnet40/modelnet40.py
@@ -72,10 +72,10 @@ def _info(self):
         citation=_CITATION,
     )
 
-  def _split_generators(self, download_manager):
+  def _split_generators(self, dl_manager):
     """Returns SplitGenerators."""
 
-    extracted_path = download_manager.download_and_extract(_URL)
+    extracted_path = dl_manager.download_and_extract(_URL)
 
     # Note: VALIDATION split was not provided by the authors
     return [