Fix tests generation

Sorry, it was my mistake
Merge pull request #7 from Bryukh/patch-3

Fix tests generation