diff --git a/src/test/java/com/googlesource/gerrit/plugins/hideactions/HideActionsActionVisitorIT.java b/src/test/java/com/googlesource/gerrit/plugins/hideactions/HideActionsActionVisitorIT.java
index d1f2740..a22b8a3 100644
--- a/src/test/java/com/googlesource/gerrit/plugins/hideactions/HideActionsActionVisitorIT.java
+++ b/src/test/java/com/googlesource/gerrit/plugins/hideactions/HideActionsActionVisitorIT.java
@@ -35,7 +35,7 @@
   public void defaultActions() throws Exception {
     PushOneCommit.Result r = createANewChangeWithTopic();
     r.assertOkStatus();
-    Map<String, ActionInfo> actions = getActions(r.getChangeId());
+    Map<String, ActionInfo> actions = gApi.changes().id(r.getChangeId()).current().actions();
     assertThat(actions).hasSize(3);
     assertThat(actions.keySet()).containsExactly("cherrypick", "rebase", "description");
   }
@@ -48,7 +48,7 @@
   public void removedAction() throws Exception {
     PushOneCommit.Result r = createANewChangeWithTopic();
     r.assertOkStatus();
-    Map<String, ActionInfo> actions = getActions(r.getChangeId());
+    Map<String, ActionInfo> actions = gApi.changes().id(r.getChangeId()).current().actions();
     assertThat(actions).hasSize(1);
     assertThat(actions.keySet()).containsExactly("description");
   }