FIX : remove keepdims for old numpy