Preserve original config when expand blocks

base_config.go

@@ -110,7 +110,7 @@ func (c *BaseConfig) expandBlock(b Block) ([]Block, error) {
 	for iterator.Next() {
 		key, value := iterator.Element()
 		newBlock := NewHclBlock(hclBlock.Block, hclBlock.wb, NewForEach(key, value))
-		nb, err := wrapBlock(c, newBlock)
+		nb, err := wrapBlock(b.Config(), newBlock)
 		if err != nil {
 			return nil, err
 		}

block.go

@@ -28,6 +28,7 @@ type Block interface {
 	PreConditionCheck(*hcl.EvalContext) ([]PreCondition, error)
 	AddressLength() int
 	CanExecutePrePlan() bool
+	Config() Config
 	getDownstreams() []Block
 	getForEach() *ForEach
 	markExpanded()

block_test.go

@@ -147,6 +147,8 @@ func (c customizedStringBlock) String() string {
 	return c.Input
 }
 
+func (c fakeBlock) Config() Config { panic("implement me") }
+
 func (c fakeBlock) Id() string {
 	panic("implement me")
 }