Bug fixing for prompt