fixed prompt showed when running test
fixed prompt showed when running test
some improvement