WordAI fix