1 files changed, 9 insertions, 21 deletions

diff --git a/bundles/org.eclipse.equinox.p2.ui/src/org/eclipse/equinox/internal/p2/ui/model/RollbackProfileElement.java b/bundles/org.eclipse.equinox.p2.ui/src/org/eclipse/equinox/internal/p2/ui/model/RollbackProfileElement.java
index 5e2e01e40..5258326e5 100644
--- a/bundles/org.eclipse.equinox.p2.ui/src/org/eclipse/equinox/internal/p2/ui/model/RollbackProfileElement.java
+++ b/bundles/org.eclipse.equinox.p2.ui/src/org/eclipse/equinox/internal/p2/ui/model/RollbackProfileElement.java
@@ -14,13 +14,9 @@ import com.ibm.icu.text.DateFormat;
 import java.util.Date;
 import org.eclipse.core.runtime.IProgressMonitor;
 import org.eclipse.core.runtime.NullProgressMonitor;
-import org.eclipse.equinox.internal.p2.ui.ProvUIMessages;
-import org.eclipse.equinox.internal.provisional.p2.core.ProvisionException;
-import org.eclipse.equinox.internal.provisional.p2.engine.IProfile;
-import org.eclipse.equinox.internal.provisional.p2.metadata.query.IQueryable;
-import org.eclipse.equinox.internal.provisional.p2.ui.ProvUIImages;
-import org.eclipse.equinox.internal.provisional.p2.ui.operations.ProvisioningUtil;
-import org.eclipse.equinox.internal.provisional.p2.ui.policy.QueryProvider;
+import org.eclipse.equinox.internal.p2.ui.*;
+import org.eclipse.equinox.p2.engine.IProfile;
+import org.eclipse.equinox.p2.query.IQueryable;
 
 /**
  * Element class for a profile snapshot
@@ -55,19 +51,16 @@ public class RollbackProfileElement extends RemoteQueriedElement {
 		return DateFormat.getDateTimeInstance(DateFormat.SHORT, DateFormat.LONG).format(new Date(timestamp));
 	}
 
+	@SuppressWarnings("rawtypes")
 	public Object getAdapter(Class adapter) {
 		if (adapter == IProfile.class)
-			try {
-				return getProfileSnapshot(new NullProgressMonitor());
-			} catch (ProvisionException e) {
-				handleException(e, ProvUIMessages.RollbackProfileElement_InvalidSnapshot);
-			}
+			return getProfileSnapshot(new NullProgressMonitor());
 		return super.getAdapter(adapter);
 	}
 
-	public IProfile getProfileSnapshot(IProgressMonitor monitor) throws ProvisionException {
+	public IProfile getProfileSnapshot(IProgressMonitor monitor) {
 		if (snapshot == null) {
-			snapshot = ProvisioningUtil.getProfile(profileId, timestamp);
+			snapshot = getProvisioningUI().getSession().getProfileRegistry().getProfile(profileId, timestamp);
 			setQueryable(snapshot);
 		}
 		return snapshot;
@@ -97,12 +90,7 @@ public class RollbackProfileElement extends RemoteQueriedElement {
 	 * (non-Javadoc)
 	 * @see org.eclipse.equinox.internal.p2.ui.model.QueriedElement#getQueryable()
 	 */
-	public IQueryable getQueryable() {
-		try {
-			return getProfileSnapshot(new NullProgressMonitor());
-		} catch (ProvisionException e) {
-			handleException(e, null);
-			return null;
-		}
+	public IQueryable<?> getQueryable() {
+		return getProfileSnapshot(new NullProgressMonitor());
 	}
 }