Changed to have an contract instead of a class when we're mocking the DAO object.