Fix broken tests.
Merge pull request #7 from ComradeDOS/master

Fixed tests