fixed tests abit
Merge pull request #49 from kunovsky/new_fixes

fixed tests abit