upping test coverage on the models