Small change to shield code