diff --git a/go.mod b/go.mod
index c3859ea2d85..f62cb699c2b 100644
--- a/go.mod
+++ b/go.mod
@@ -306,6 +306,6 @@ require (
 
 replace github.com/ethereum/go-ethereum v1.13.15 => github.com/taikoxyz/taiko-geth v0.0.0-20240504072040-7e1b8b65a3f8
 
-replace github.com/ethereum-optimism/optimism v1.7.4 => github.com/taikoxyz/optimism v0.0.0-20240508063202-db34e8c760ea
+replace github.com/ethereum-optimism/optimism v1.7.4 => github.com/taikoxyz/optimism v0.0.0-20240608043351-5c8ed8012bec
 
 replace github.com/uber/jaeger-client-go => github.com/uber/jaeger-client-go v2.25.0+incompatible
diff --git a/go.sum b/go.sum
index 66743bc646c..a580a86cb06 100644
--- a/go.sum
+++ b/go.sum
@@ -1121,8 +1121,8 @@ github.com/swaggo/swag v1.16.3/go.mod h1:DImHIuOFXKpMFAQjcC7FG4m3Dg4+QuUgUzJmKjI
 github.com/syndtr/goleveldb v1.0.1-0.20210819022825-2ae1ddf74ef7/go.mod h1:q4W45IWZaF22tdD+VEXcAWRA037jwmWEB5VWYORlTpc=
 github.com/syndtr/goleveldb v1.0.1-0.20220721030215-126854af5e6d h1:vfofYNRScrDdvS342BElfbETmL1Aiz3i2t0zfRj16Hs=
 github.com/syndtr/goleveldb v1.0.1-0.20220721030215-126854af5e6d/go.mod h1:RRCYJbIwD5jmqPI9XoAFR0OcDxqUctll6zUj/+B4S48=
-github.com/taikoxyz/optimism v0.0.0-20240508063202-db34e8c760ea h1:+oLT2sxS+KLgHZVW4toyKavAMbgjYkFOiL0YpciATnQ=
-github.com/taikoxyz/optimism v0.0.0-20240508063202-db34e8c760ea/go.mod h1:bkiRWtDE4T9nXmaK5NBKC/buX5GnyJ6VKozGFwiwKxs=
+github.com/taikoxyz/optimism v0.0.0-20240608043351-5c8ed8012bec h1:QwQ+16rm8/Me6KQqLbPBrZhq4xHwzU1Cn3vlJKqROBA=
+github.com/taikoxyz/optimism v0.0.0-20240608043351-5c8ed8012bec/go.mod h1:bkiRWtDE4T9nXmaK5NBKC/buX5GnyJ6VKozGFwiwKxs=
 github.com/taikoxyz/taiko-geth v0.0.0-20240504072040-7e1b8b65a3f8 h1:z4juQ4Nyp2T836JTCNC8t3vrbr0K9v2pPUV/ir2dy9s=
 github.com/taikoxyz/taiko-geth v0.0.0-20240504072040-7e1b8b65a3f8/go.mod h1:nqByouVW0a0qx5KKgvYgoXba+pYEHznAAQp6LhZilgM=
 github.com/tarm/serial v0.0.0-20180830185346-98f6abe2eb07/go.mod h1:kDXzergiv9cbyO7IOYJZWg1U88JhDg3PB6klq9Hg2pA=
diff --git a/packages/taiko-client/proposer/prover_selector/eth_fee_eoa_selector_test.go b/packages/taiko-client/proposer/prover_selector/eth_fee_eoa_selector_test.go
index c7fa58e8cbe..aa156d04cc1 100644
--- a/packages/taiko-client/proposer/prover_selector/eth_fee_eoa_selector_test.go
+++ b/packages/taiko-client/proposer/prover_selector/eth_fee_eoa_selector_test.go
@@ -39,8 +39,8 @@ func (s *ProverSelectorTestSuite) SetupTest() {
 		s.RPCClient,
 		crypto.PubkeyToAddress(l1ProposerPrivKey.PublicKey),
 		common.HexToAddress(os.Getenv("TAIKO_L1_ADDRESS")),
-		common.HexToAddress(os.Getenv("ASSIGNMENT_HOOK_ADDRESS")),
 		common.HexToAddress(os.Getenv("PROVER_SET_ADDRESS")),
+		common.HexToAddress(os.Getenv("ASSIGNMENT_HOOK_ADDRESS")),
 		[]encoding.TierFee{},
 		common.Big2,
 		[]*url.URL{s.ProverEndpoints[0]},
@@ -60,9 +60,9 @@ func (s *ProverSelectorTestSuite) TestProverAssignProver() {
 		{Tier: encoding.TierOptimisticID, Fee: common.Big256},
 		{Tier: encoding.TierSgxID, Fee: common.Big256},
 	}, testutils.RandomHash())
+	s.Nil(err)
 	s.NotEmpty(sig)
 	s.Equal(fee.Cmp(common.Big32), 1)
-	s.Nil(err)
 }
 
 func TestProverSelectorTestSuite(t *testing.T) {