AI-406 : test fix