Fix test_ufuncs