diff --git a/numpy/lib/shape_base.py b/numpy/lib/shape_base.py
index b2beef0a8944..26c2aab0496f 100644
--- a/numpy/lib/shape_base.py
+++ b/numpy/lib/shape_base.py
@@ -424,7 +424,7 @@ def array_split(ary, indices_or_sections, axis=0):
     # This "kludge" was introduced here to replace arrays shaped (0, 10)
     # or similar with an array shaped (0,).
     # There seems no need for this, so give a FutureWarning to remove later.
-    if any(arr.size == 0 and arr.ndim != 1 for arr in sub_arys):
+    if sub_arys[-1].size == 0 and sub_arys[-1].ndim != 1:
         warnings.warn("in the future np.array_split will retain the shape of "
                       "arrays with a zero size, instead of replacing them by "
                       "`array([])`, which always has a shape of (0,).",
diff --git a/numpy/lib/tests/test_shape_base.py b/numpy/lib/tests/test_shape_base.py
index 8ab72b9f938b..3f2d8d5b4a01 100644
--- a/numpy/lib/tests/test_shape_base.py
+++ b/numpy/lib/tests/test_shape_base.py
@@ -111,15 +111,6 @@ def test_integer_split_2D_rows(self):
         compare_results(res, desired)
         assert_(a.dtype.type is res[-1].dtype.type)
 
-        # Same thing for manual splits:
-        res = assert_warns(FutureWarning, array_split, a, [0, 1, 2], axis=0)
-
-        # After removing the FutureWarning, the last should be zeros((0, 10))
-        desired = [np.array([]), np.array([np.arange(10)]),
-                   np.array([np.arange(10)])]
-        compare_results(res, desired)
-        assert_(a.dtype.type is res[-1].dtype.type)
-
     def test_integer_split_2D_cols(self):
         a = np.array([np.arange(10), np.arange(10)])
         res = array_split(a, 3, axis=-1)