Fix shader bugs