fix prompt bug