Fixed benchmark tests
Merge pull request #2 from joonas/fix_benchmark_tests

Fixed benchmark tests