BUG: wrong import
TST: Fix tests under numpy 1.8.0