(IDEA) Checking the profession of a villager

At the time of development on one of my mods, I ran into a problem - I needed to check if a villager had a certain profession. As it turned out, there is no procedural block in MCreator responsible for this. I would like you to do that. After a while, I managed to find a way to solve this problem (using custom code), but it does not work with the custom professions of the villagers. To make it easier for you, I'll leave the code:
((Villager) entity).getVillagerData().getProfession() == VillagerProfession.PROFESSIONNAME