Fix YUI tests