fixed failing test
Merge pull request #349 from glee23/master

Fixed failing test