fixed test function
Merge pull request #121 from giulianoriccio/master

fixed test function