BUG: provide better ndarray compat