Fix shader bug