diff --git a/nodes.py b/nodes.py
index e57f1ff..b829502 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1281,7 +1281,6 @@ class Hy3DBPT:
                 "enable_bpt": ("BOOLEAN", {"default": True}),
                 "temperature": ("FLOAT", {"default": 0.5}),
                 "batch_size": ("INT", {"default": 1}),
-                "with_normal": ("BOOLEAN", {"default": True})
             },
         }
 
@@ -1290,16 +1289,16 @@ class Hy3DBPT:
     FUNCTION = "bpt"
     CATEGORY = "Hunyuan3DWrapper"
     DESCRIPTION = "BPT the mesh using bpt: https://github.com/whaohan/bpt"
+    
+    def bpt(self, trimesh, enable_bpt, temperature, batch_size):
 
-    def bpt(self, trimesh, enable_bpt, with_normal, temperature, batch_size):
-        from .hy3dgen.shapegen import BptMesh
         new_mesh = trimesh.copy()
-
         if enable_bpt:
-            new_mesh = BptMesh()(new_mesh, with_normal=with_normal, temperature=temperature, batch_size=batch_size)
-            
-        mm.unload_all_models()
-        mm.soft_empty_cache()
+            from .hy3dgen.shapegen.postprocessors import BptMesh
+            new_mesh = BptMesh()(new_mesh, with_normal=True, temperature=temperature, batch_size=batch_size)
+            mm.unload_all_models()
+            mm.soft_empty_cache()
+
         return (new_mesh, )
             
 class Hy3DGetMeshPBRTextures: