Use check_array and remove shape check