diff --git a/backend/tests/config.py b/backend/tests/config.py
index c9b6bcfc..2b89a4a1 100644
--- a/backend/tests/config.py
+++ b/backend/tests/config.py
@@ -157,6 +157,5 @@ class BaseProposalCreatorConfig(BaseUserConfig):
         # 3. fake a confirmation
         contribution.confirm(tx_id='tx', amount=str(PROPOSAL_STAKING_AMOUNT.normalize()))
         db.session.add(contribution)
-        db.session.commit()
-        contribution = self.proposal.get_staking_contribution(self.user.id)
-        return contribution
+        db.session.flush()
+        self.proposal.set_pending_when_ready()
diff --git a/backend/tests/proposal/test_api.py b/backend/tests/proposal/test_api.py
index dff7a2c9..3adc28bc 100644
--- a/backend/tests/proposal/test_api.py
+++ b/backend/tests/proposal/test_api.py
@@ -139,7 +139,7 @@ class TestProposalAPI(BaseProposalCreatorConfig):
         self.stake_proposal()
         resp = self.app.get(f"/api/v1/proposals/{self.proposal.id}/stake")
         print(resp)
-        self.assert404(resp)
+        self.assert400(resp)
 
     # /publish
     @patch('requests.get', side_effect=mock_blockchain_api_requests)