Fixing bug in test
Merge pull request #3 from travisfischer/master

Fix for bug