diff --git a/pandas/tests/groupby/test_apply.py b/pandas/tests/groupby/test_apply.py
index a4a0f863cf6b3..da6e7c6851667 100644
--- a/pandas/tests/groupby/test_apply.py
+++ b/pandas/tests/groupby/test_apply.py
@@ -968,10 +968,9 @@ def test_apply_function_with_indexing_return_column():
     df = DataFrame(
         {
             "foo1": ["one", "two", "two", "three", "one", "two"],
-            "foo2": np.random.randn(6),
+            "foo2": [1, 2, 3, 4, 5, 6],
         }
     )
     result = df.groupby("foo1", as_index=False).apply(lambda x: x.mean())
     expected = df.groupby("foo1", as_index=False).mean()
     tm.assert_frame_equal(result, expected)
-    assert "foo1" in result.columns