diff --git a/flyteplugins/go/tasks/plugins/array/catalog.go b/flyteplugins/go/tasks/plugins/array/catalog.go
index e3ca301f76..1161fa306a 100644
--- a/flyteplugins/go/tasks/plugins/array/catalog.go
+++ b/flyteplugins/go/tasks/plugins/array/catalog.go
@@ -110,7 +110,7 @@ func DetermineDiscoverability(ctx context.Context, tCtx core.TaskExecutionContex
 
 		indexLookup := CatalogBitsetToLiteralCollection(cachedResults, resp.GetResultsSize())
 		// TODO: Is the right thing to use?  Haytham please take a look
-		indexLookupPath, err := ioutils.GetIndexLookupPath(ctx, tCtx.DataStore(), tCtx.OutputWriter().GetOutputPrefixPath())
+		indexLookupPath, err := ioutils.GetIndexLookupPath(ctx, tCtx.DataStore(), tCtx.InputReader().GetInputPrefixPath())
 		if err != nil {
 			return state, err
 		}