Fixed the prompt bug