diff --git a/profiles/killbill/src/test/java/org/killbill/billing/jaxrs/TestBundle.java b/profiles/killbill/src/test/java/org/killbill/billing/jaxrs/TestBundle.java
index 84bc268..e9bc2be 100644
--- a/profiles/killbill/src/test/java/org/killbill/billing/jaxrs/TestBundle.java
+++ b/profiles/killbill/src/test/java/org/killbill/billing/jaxrs/TestBundle.java
@@ -140,7 +140,7 @@ public class TestBundle extends TestJaxrsBase {
assertEquals(bundle.getExternalKey(), bundleExternalKey);
final BlockingState blockingState = new BlockingState(bundle.getBundleId(), "block", "service", false, true, true, clock.getToday(DateTimeZone.forID(accountJson.getTimeZone())), BlockingStateType.SUBSCRIPTION_BUNDLE, null);
- killBillClient.blockBundle(bundle.getBundleId(), blockingState, createdBy, reason, comment);
+ killBillClient.setBlockingState(bundle.getBundleId(), blockingState, createdBy, reason, comment);
final Subscription subscription = killBillClient.getSubscription(entitlement.getSubscriptionId());
assertEquals(subscription.getState(), EntitlementState.BLOCKED);
@@ -148,7 +148,7 @@ public class TestBundle extends TestJaxrsBase {
clock.addDays(1);
final BlockingState unblockingState = new BlockingState(bundle.getBundleId(), "unblock", "service", false, false, false, clock.getToday(DateTimeZone.forID(accountJson.getTimeZone())), BlockingStateType.SUBSCRIPTION_BUNDLE, null);
- killBillClient.blockBundle(bundle.getBundleId(), unblockingState, createdBy, reason, comment);
+ killBillClient.setBlockingState(bundle.getBundleId(), unblockingState, createdBy, reason, comment);
final Subscription subscription2 = killBillClient.getSubscription(entitlement.getSubscriptionId());
assertEquals(subscription2.getState(), EntitlementState.ACTIVE);