Fix recursive prompt

There you go! Just needed to split them in two functions.