diff --git a/plugin/trino-iceberg/src/main/java/io/trino/plugin/iceberg/RollbackToSnapshotProcedure.java b/plugin/trino-iceberg/src/main/java/io/trino/plugin/iceberg/RollbackToSnapshotProcedure.java
index 3008065f7cc1..af22fe183979 100644
--- a/plugin/trino-iceberg/src/main/java/io/trino/plugin/iceberg/RollbackToSnapshotProcedure.java
+++ b/plugin/trino-iceberg/src/main/java/io/trino/plugin/iceberg/RollbackToSnapshotProcedure.java
@@ -74,7 +74,7 @@ public void rollbackToSnapshot(ConnectorSession clientSession, String schema, St
         try (ThreadContextClassLoader ignored = new ThreadContextClassLoader(classLoader)) {
             SchemaTableName schemaTableName = new SchemaTableName(schema, table);
             Table icebergTable = catalogFactory.create(clientSession.getIdentity()).loadTable(clientSession, schemaTableName);
-            icebergTable.rollback().toSnapshotId(snapshotId).commit();
+            icebergTable.manageSnapshots().setCurrentSnapshot(snapshotId).commit();
         }
     }
 }