fix prompt behaviour