fixed test and added pry