fix prompt function