Fixed already made tests
Merge pull request #29 from Nigatsu/feature/test-fix

Feature/test fix