diff --git a/presto-iceberg/src/main/java/com/facebook/presto/iceberg/IcebergAbstractMetadata.java b/presto-iceberg/src/main/java/com/facebook/presto/iceberg/IcebergAbstractMetadata.java
index 4d30b08e10239..01e1b161ed4c7 100644
--- a/presto-iceberg/src/main/java/com/facebook/presto/iceberg/IcebergAbstractMetadata.java
+++ b/presto-iceberg/src/main/java/com/facebook/presto/iceberg/IcebergAbstractMetadata.java
@@ -200,10 +200,10 @@ protected final Table getIcebergTable(ConnectorSession session, SchemaTableName
     protected abstract void unregisterTable(ConnectorSession clientSession, SchemaTableName schemaTableName);
 
     /**
-     * This class implements the default implementation for getTableLayouts which will be used in the case of a Java Worker
+     * This class implements the default implementation for getTableLayoutForConstraint which will be used in the case of a Java Worker
      */
     @Override
-    public List<ConnectorTableLayoutResult> getTableLayouts(
+    public ConnectorTableLayoutResult getTableLayoutForConstraint(
             ConnectorSession session,
             ConnectorTableHandle table,
             Constraint<ColumnHandle> constraint,
@@ -233,7 +233,7 @@ public List<ConnectorTableLayoutResult> getTableLayouts(
                         .setPartitions(Optional.empty())
                         .setTable(handle)
                         .build());
-        return ImmutableList.of(new ConnectorTableLayoutResult(layout, constraint.getSummary()));
+        return new ConnectorTableLayoutResult(layout, constraint.getSummary());
     }
 
     public static Subfield toSubfield(ColumnHandle columnHandle)