Fix prompt not updating correctly