Fix shader issue