diff --git a/olcCodeJam2023Entry/Scenario.cpp b/olcCodeJam2023Entry/Scenario.cpp
index 7c9bf7c..d18a0f6 100644
--- a/olcCodeJam2023Entry/Scenario.cpp
+++ b/olcCodeJam2023Entry/Scenario.cpp
@@ -143,6 +143,7 @@ void Scenario::RunAI(Resources&enemy_resources,std::vector<std::shared_ptr<Colle
 								if(enemy_resources.range>0){enemy_resources.range=std::max(0,enemy_resources.range-totalCost);}else
 								{enemy_resources.procedure=std::max(0,enemy_resources.procedure-totalCost);
 								}
+					availableMemory-=totalCost;
 				}
 			}
 			if(!u->IsFriendly()&&u->IsPlatform()){
@@ -174,6 +175,7 @@ void Scenario::RunAI(Resources&enemy_resources,std::vector<std::shared_ptr<Colle
 								if(enemy_resources.range>0){enemy_resources.range=std::max(0,enemy_resources.range-totalCost);}else
 								{enemy_resources.procedure=std::max(0,enemy_resources.procedure-totalCost);
 								}
+					availableMemory-=totalCost;
 				}
 			}
 			switch(flags.difficulty){
@@ -650,6 +652,7 @@ void Stage4::Update(){
 				state=6;
 				box.SetVisible(false);
 				flags.playerInControl=true;
+				flags.flashMemoryBar=false;
 			}
 		}break;
 		case 6:{
diff --git a/olcCodeJam2023Entry/pge.wasm b/olcCodeJam2023Entry/pge.wasm
index 5434803..0dbe41f 100644
Binary files a/olcCodeJam2023Entry/pge.wasm and b/olcCodeJam2023Entry/pge.wasm differ