Added dependency as model has moved into a new module