Fix test and add byebug