Updated test to reflect most recent changes.