Fix prompt problem