A little refactoring in the test.
Merge pull request #1 from marcusesa/refactoring-test

A little refactoring in the test.