Fix for modules. There is a second parameter for protected classes.