diff --git a/packages/taiko-client/prover/server/api.go b/packages/taiko-client/prover/server/api.go
index 62c0c81a93..5e305ad4bb 100644
--- a/packages/taiko-client/prover/server/api.go
+++ b/packages/taiko-client/prover/server/api.go
@@ -140,7 +140,7 @@ func (s *ProverServer) CreateAssignment(c echo.Context) error {
 	if ok, err = rpc.CheckProverBalance(
 		c.Request().Context(),
 		s.rpc,
-		s.proverAddress,
+		prover,
 		s.assignmentHookAddress,
 		s.livenessBond,
 	); err != nil {