Updated mock model to use a public attribute