fix numpy conditional import