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

I don't know how this is working without the second parameter...