Update test case to match new example
Merge pull request #564 from maxsnew/fix-test

Update test case to match new example