Improve unit test by testing underlying model calls.