Updated test to reflect design changes.