Fix tests generation

Sorry, it was my mistake