diff --git a/sdk/src/test/java/com/google/cloud/dataflow/sdk/io/CountingInputTest.java b/sdk/src/test/java/com/google/cloud/dataflow/sdk/io/CountingInputTest.java
index 948a892b58a9..1daadc7e2b7f 100644
--- a/sdk/src/test/java/com/google/cloud/dataflow/sdk/io/CountingInputTest.java
+++ b/sdk/src/test/java/com/google/cloud/dataflow/sdk/io/CountingInputTest.java
@@ -33,10 +33,13 @@
 import org.joda.time.Instant;
 import org.junit.Test;
 import org.junit.experimental.categories.Category;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
 
 /**
  * Tests for {@link CountingInput}.
  */
+@RunWith(JUnit4.class)
 public class CountingInputTest {
   public static void addCountingAsserts(PCollection<Long> input, long numElements) {
     // Count == numElements