Fix tests script
Merge pull request #11 from alexeyvo/v2

Fix tests script