Fix tests, require 'yaml'

Some tests fail here, because yaml is not imported.