Fix model attribute name in test.

[#76962004] https://www.pivotaltracker.com/story/show/76962004

Signed-off-by: Madhura Bhave <mbhave@gopivotal.com>