Fix for toml test.

ten-protocol · Sep 19, 2023 · 7b1d6f2 · 7b1d6f2
1 parent 401c0cd
commit 7b1d6f2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/go/enclave/container/cli.go b/go/enclave/container/cli.go
@@ -40,6 +40,7 @@ type EnclaveConfigToml struct {
 	MaxRollupSize             uint64
 	GasPaymentAddress         string
 	BaseFee                   uint64
+	GasLimit                  uint64
 }
 
 // ParseConfig returns a config.EnclaveConfig based on either the file identified by the `config` flag, or the flags
@@ -73,6 +74,7 @@ func ParseConfig() (*config.EnclaveConfig, error) {
 	maxRollupSize := flag.Uint64(maxRollupSizeName, cfg.MaxRollupSize, flagUsageMap[maxRollupSizeName])
 	baseFee := flag.Uint64("l2BaseFee", cfg.BaseFee.Uint64(), "")
 	coinbaseAddress := flag.String("l2Coinbase", cfg.GasPaymentAddress.Hex(), "")
+	gasLimit := flag.Uint64("l2GasLimit", cfg.GasLimit.Uint64(), "")
 
 	flag.Parse()
 
@@ -109,6 +111,7 @@ func ParseConfig() (*config.EnclaveConfig, error) {
 	cfg.MaxRollupSize = *maxRollupSize
 	cfg.BaseFee = big.NewInt(0).SetUint64(*baseFee)
 	cfg.GasPaymentAddress = gethcommon.HexToAddress(*coinbaseAddress)
+	cfg.GasLimit = big.NewInt(0).SetUint64(*gasLimit)
 
 	return cfg, nil
 }