fix a test or two