Fixed a possible bug with shields