fixed test refactoring
Merge pull request #66 from sprankhub/fix

fixed test refactoring