Remove pry from test