Updated shader to support engine update.