diff --git a/pytorch3d/implicitron/models/overfit_model.py b/pytorch3d/implicitron/models/overfit_model.py
index 52854d05..40ee5a1b 100644
--- a/pytorch3d/implicitron/models/overfit_model.py
+++ b/pytorch3d/implicitron/models/overfit_model.py
@@ -373,7 +373,7 @@ class OverfitModel(ImplicitronModelBase):  # pyre-ignore: 13
             self.implicit_function
         ]
         if self.coarse_implicit_function is not None:
-            implicit_functions += [self.coarse_implicit_function]
+            implicit_functions = [self.coarse_implicit_function, self.implicit_function]
 
         if self.global_encoder is not None:
             global_code = self.global_encoder(  # pyre-fixme[29]
diff --git a/tests/implicitron/models/test_overfit_model.py b/tests/implicitron/models/test_overfit_model.py
index 15a6a6c6..8012e214 100644
--- a/tests/implicitron/models/test_overfit_model.py
+++ b/tests/implicitron/models/test_overfit_model.py
@@ -109,9 +109,9 @@ class TestOverfitModel(unittest.TestCase):
 
         # Adapt the mapping from generic model to overfit model
         mapping_om_from_gm = {
-            key.replace("_implicit_functions.0._fn", "implicit_function").replace(
-                "_implicit_functions.1._fn", "coarse_implicit_function"
-            ): val
+            key.replace(
+                "_implicit_functions.0._fn", "coarse_implicit_function"
+            ).replace("_implicit_functions.1._fn", "implicit_function"): val
             for key, val in generic_model.state_dict().items()
         }
         # Copy parameters from generic_model to overfit_model