set_prompt bug fix