Tests fix
Merge pull request #5 from norzechowicz/tests

Tests fix