diff --git a/auto_round/autoround.py b/auto_round/autoround.py
index ff388911..361c820f 100644
--- a/auto_round/autoround.py
+++ b/auto_round/autoround.py
@@ -338,11 +338,11 @@ def set_layerwise_config(self, weight_config):
         Returns:
         None
         """
-        layers_inblocks = get_layer_names_in_block(self.model, self.supported_types)
+        layers_in_blocks = get_layer_names_in_block(self.model, self.supported_types)
         for n, m in self.model.named_modules():
             if not isinstance(m, tuple(self.supported_types)):
                 continue
-            if n not in weight_config.keys() and n in layers_inblocks:
+            if n not in weight_config.keys() and n in layers_in_blocks:
                 weight_config[n] = {}
                 weight_config[n]["data_type"] = self.data_type
                 weight_config[n]["bits"] = self.bits
diff --git a/docs/imgs/autoround_overview.png b/docs/imgs/autoround_overview.png
index ac13df61..97d66c68 100644
Binary files a/docs/imgs/autoround_overview.png and b/docs/imgs/autoround_overview.png differ