Fix prompt of shell in the test