diff --git a/gorge.js b/gorge.js
index f7dddaf..03c0415 100644
--- a/gorge.js
+++ b/gorge.js
@@ -301,6 +301,9 @@ function spend(cost) {
 }
 
 function switchShowOwnedUpgrades() {
+
+  initializeUpgradeStates();
+  
   if (showOwnedUpgrades) {
     document.querySelector("#upgrades").innerText = "Upgrades";
   } else {
@@ -476,6 +479,11 @@ const states = {};
 // we can just check if they've been handled already
 
 function initializeStates() {
+  initializeBuildingStates();
+  initializeUpgradeStates();
+}
+
+function initializeBuildingStates() {
   const buildingStates = {};
 
   Object.keys(buildings).forEach(key => {
@@ -486,7 +494,9 @@ function initializeStates() {
   });
 
   states.buildings = buildingStates;
+}
 
+function initializeUpgradeStates() {
   const upgradeStates = {};
 
   Object.keys(upgrades).forEach(key => {
@@ -498,6 +508,8 @@ function initializeStates() {
 
   states.upgrades = upgradeStates;
 }
+
+
 function unlockAtStart() {
   unlockBuilding("micro");