Fix a wrong variable name in the `Model` class