- Fix the entity shader hook