Change generation step in test