FIX: Extra np and nibabel fixes