fixed the test file

Hello Jordan,

just in case you might be still interested, 
I have slightly modified your code to make the tests run 
in the current setting.

Cheers
Sven