Fix test case
Fix ndim test