diff --git a/python/tasks/mlb_player.py b/python/tasks/mlb_player.py
index 91e147b..964a892 100644
--- a/python/tasks/mlb_player.py
+++ b/python/tasks/mlb_player.py
@@ -195,7 +195,7 @@ def kernel_4(
         }
     """
 
-    t1 = numpy.where(o_3['t5']['events']['gamePk'] == 634280)
+    t1 = numpy.where(o_3['t5']['events']['gamePk'] == 634280)[0]
     t2 = o_3['t5']['events'].sel(index=t1)
     t3 = o_3['t5']['games'].to_dataframe().iloc[-1].to_dict()
     assert t3['gamePk'] == 634280