diff --git a/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/SplittableParDo.java b/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/SplittableParDo.java
index 8d8da216afac..b581eecf4140 100644
--- a/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/SplittableParDo.java
+++ b/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/SplittableParDo.java
@@ -415,7 +415,6 @@ private static class PairWithRestrictionFn<InputT, OutputT, RestrictionT>
     @Setup
     public void setup() {
       invoker = DoFnInvokers.invokerFor(fn);
-      invoker.invokeSetup();
     }
 
     @ProcessElement
@@ -423,12 +422,6 @@ public void processElement(ProcessContext context) {
       context.output(
           KV.of(context.element(), invoker.invokeGetInitialRestriction(context.element())));
     }
-
-    @Teardown
-    public void tearDown() {
-      invoker.invokeTeardown();
-      invoker = null;
-    }
   }
 
   /** Splits the restriction using the given {@link SplitRestriction} method. */
@@ -446,7 +439,6 @@ private static class SplitRestrictionFn<InputT, RestrictionT>
     @Setup
     public void setup() {
       invoker = DoFnInvokers.invokerFor(splittableFn);
-      invoker.invokeSetup();
     }
 
     @ProcessElement
@@ -467,11 +459,5 @@ public void outputWithTimestamp(RestrictionT part, Instant timestamp) {
             }
           });
     }
-
-    @Teardown
-    public void tearDown() {
-      invoker.invokeTeardown();
-      invoker = null;
-    }
   }
 }
diff --git a/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/SplittableDoFnTest.java b/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/SplittableDoFnTest.java
index fe33b1ab5b8c..b7f0c10d046a 100644
--- a/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/SplittableDoFnTest.java
+++ b/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/SplittableDoFnTest.java
@@ -513,19 +513,18 @@ private enum State {
 
     private State state = State.BEFORE_SETUP;
 
+    @ProcessElement
+    public void processElement(ProcessContext c, OffsetRangeTracker tracker) {
+      assertEquals(State.INSIDE_BUNDLE, state);
+      assertTrue(tracker.tryClaim(0L));
+      c.output(c.element());
+    }
+
     @GetInitialRestriction
     public OffsetRange getInitialRestriction(String value) {
-      assertEquals(State.OUTSIDE_BUNDLE, state);
       return new OffsetRange(0, 1);
     }
 
-    @SplitRestriction
-    public void splitRestriction(
-        String value, OffsetRange range, OutputReceiver<OffsetRange> receiver) {
-      assertEquals(State.OUTSIDE_BUNDLE, state);
-      receiver.output(range);
-    }
-
     @Setup
     public void setUp() {
       assertEquals(State.BEFORE_SETUP, state);
@@ -538,13 +537,6 @@ public void startBundle() {
       state = State.INSIDE_BUNDLE;
     }
 
-    @ProcessElement
-    public void processElement(ProcessContext c, OffsetRangeTracker tracker) {
-      assertEquals(State.INSIDE_BUNDLE, state);
-      assertTrue(tracker.tryClaim(0L));
-      c.output(c.element());
-    }
-
     @FinishBundle
     public void finishBundle() {
       assertEquals(State.INSIDE_BUNDLE, state);
@@ -561,9 +553,12 @@ public void tearDown() {
   @Test
   @Category({ValidatesRunner.class, UsesSplittableParDo.class})
   public void testLifecycleMethods() throws Exception {
+
     PCollection<String> res =
         p.apply(Create.of("a", "b", "c")).apply(ParDo.of(new SDFWithLifecycle()));
+
     PAssert.that(res).containsInAnyOrder("a", "b", "c");
+
     p.run();
   }