fix prompt logic