Fix store upgrades integration

[anilcse]

Description

Checklist

• Targeted PR against correct branch.
• Linked to Github issue with discussion and accepted design OR link to spec that describes this work.
• Wrote unit tests.
• Wrote integration tests (simulation & CLI).
• Updated the documentation.
• Added an entry to the CHANGELOG.md file.
• Re-reviewed Files changed in the Github PR explorer.

[codecov]

Codecov Report

❗ No coverage uploaded for pull request base (riccardo/with-module@3256d8b). Click here to learn what that means.
The diff coverage is n/a.

@@                   Coverage Diff                   @@
##             riccardo/with-module     #337   +/-   ##
=======================================================
  Coverage                        ?   80.07%           
=======================================================
  Files                           ?      102           
  Lines                           ?     3909           
  Branches                        ?        0           
=======================================================
  Hits                            ?     3130           
  Misses                          ?      662           
  Partials                        ?      117           

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 3256d8b...37d291a. Read the comment docs.