ENH: Better test