Fix a custom shader issue