Partially revert changes made to files.tests.test_models in aa43c6ba98ee0fa5154c90ad8610e4a4877f8298

The test was right before, we just needed to set the return value of
the mock earlier.