ENH: fixed tests