I've noticed the same problem. Exiting out of the shop and re-entering after buying several upgrades seems to be the best workaround for now.