diff --git a/hy3dgen/shapegen/models/autoencoders/surface_extractors.py b/hy3dgen/shapegen/models/autoencoders/surface_extractors.py
index f4d8f63..9e99ca4 100644
--- a/hy3dgen/shapegen/models/autoencoders/surface_extractors.py
+++ b/hy3dgen/shapegen/models/autoencoders/surface_extractors.py
@@ -66,6 +66,7 @@ class SurfaceExtractor:
 
 class MCSurfaceExtractor(SurfaceExtractor):
     def run(self, grid_logit, *, mc_level, bounds, octree_resolution, **kwargs):
+        print("MC Surface Extractor")
         vertices, faces, normals, _ = measure.marching_cubes(
             grid_logit.cpu().numpy(),
             mc_level,
@@ -79,6 +80,7 @@ class MCSurfaceExtractor(SurfaceExtractor):
 class DMCSurfaceExtractor(SurfaceExtractor):
     def run(self, grid_logit, *, octree_resolution, **kwargs):
         device = grid_logit.device
+        print("DMC Surface Extractor")
         if not hasattr(self, 'dmc'):
             try:
                 from diso import DiffDMC
diff --git a/nodes.py b/nodes.py
index e7f6821..0df9be8 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1213,7 +1213,9 @@ class Hy3DVAEDecode:
 
         vae.to(device)
 
-        vae.enable_flashvdm_decoder(enabled=enable_flash_vdm)
+        vae.enable_flashvdm_decoder(
+            enabled=enable_flash_vdm,
+            mc_algo=mc_algo,)
         
         latents = 1. / vae.scale_factor * latents
         latents = vae(latents)