fix bug of test code
add gem pry