Small refactoring in the models protection