diff --git a/src/open_samus_returns_rando/files/custom/scenario.lua b/src/open_samus_returns_rando/files/custom/scenario.lua
index ac15515..0f1cc21 100644
--- a/src/open_samus_returns_rando/files/custom/scenario.lua
+++ b/src/open_samus_returns_rando/files/custom/scenario.lua
@@ -28,7 +28,7 @@ function Scenario._UpdateProgressiveItemModels()
           local pickup = Game.GetEntity(name)
           pickup.MODELUPDATER.sModelAlias = model.alias
           -- Rotate Wave Beam and High Jump Boots to face right
-          if index == 1 then
+          if model.alias == "powerup_wavebeam" or model.alias == "powerup_highjumpboots" then
             pickup.vAng = V3D(0, 1.5, 0)
           end
           break