Fix tests, missing require 'yaml'