diff --git a/torchinfo/layer_info.py b/torchinfo/layer_info.py
index 3bf2d27..9a85bf8 100644
--- a/torchinfo/layer_info.py
+++ b/torchinfo/layer_info.py
@@ -39,7 +39,7 @@ def __init__(
         self.class_name = (
             str(module.original_name)
             if isinstance(module, ScriptModule)
-            else module.__class__.__name__
+            else module._get_name()
         )
         # {layer name: {col_name: value_for_row}}
         self.inner_layers: dict[str, dict[ColumnSettings, Any]] = {}