improves code style of test.
Merge pull request #2 from dbernstein/master

improves code style of test.