fix generated test code (first shot)