Fixed a problem with the custom shader