Fix shader reload