I suggest making it require holding the interact key to pick up or drop the shield, and also adding functionality to equip the shield instantly when casting the ability by holding the ability key. There's also a bug where you don't gain energy from energy pads when the shield has been picked up.