diff --git a/samza-test/src/main/java/org/apache/samza/test/framework/TestRunner.java b/samza-test/src/main/java/org/apache/samza/test/framework/TestRunner.java
index 6d6613c98a..fe8581b7b7 100644
--- a/samza-test/src/main/java/org/apache/samza/test/framework/TestRunner.java
+++ b/samza-test/src/main/java/org/apache/samza/test/framework/TestRunner.java
@@ -33,7 +33,6 @@
 import org.apache.samza.SamzaException;
 import org.apache.samza.application.LegacyTaskApplication;
 import org.apache.samza.application.SamzaApplication;
-import org.apache.samza.application.StreamApplication;
 import org.apache.samza.config.Config;
 import org.apache.samza.config.JobConfig;
 import org.apache.samza.config.JobCoordinatorConfig;
@@ -63,7 +62,6 @@
 import org.apache.samza.test.framework.system.InMemorySystemDescriptor;
 import org.junit.Assert;
 
-
 /**
  * TestRunner provides APIs to set up integration tests for a Samza application.
  * Running mode for test is Single container mode
@@ -102,7 +100,7 @@ private TestRunner() {
 
   /**
    * Constructs a new {@link TestRunner} from following components
-   * @param taskClass represent a class containing Samza job logic extending either {@link StreamTask} or {@link AsyncStreamTask}
+   * @param taskClass containing Samza job logic extending either {@link StreamTask} or {@link AsyncStreamTask}
    */
   private TestRunner(Class taskClass) {
     this();
@@ -113,9 +111,9 @@ private TestRunner(Class taskClass) {
 
   /**
    * Constructs a new {@link TestRunner} from following components
-   * @param app samza job implementing {@link StreamApplication}
+   * @param app a {@link SamzaApplication}
    */
-  private TestRunner(StreamApplication app) {
+  private TestRunner(SamzaApplication app) {
     this();
     Preconditions.checkNotNull(app);
     this.app = app;
@@ -134,11 +132,11 @@ public static TestRunner of(Class taskClass) {
   }
 
   /**
-   * Creates an instance of {@link TestRunner} for High Level/Fluent Samza Api
-   * @param app samza job implementing {@link StreamApplication}
+   * Creates an instance of {@link TestRunner} for a {@link SamzaApplication}
+   * @param app a {@link SamzaApplication}
    * @return this {@link TestRunner}
    */
-  public static TestRunner of(StreamApplication app) {
+  public static TestRunner of(SamzaApplication app) {
     Preconditions.checkNotNull(app);
     return new TestRunner(app);
   }
diff --git a/samza-test/src/test/java/org/apache/samza/test/framework/AsyncStreamTaskIntegrationTest.java b/samza-test/src/test/java/org/apache/samza/test/framework/AsyncStreamTaskIntegrationTest.java
index 581b1c33c2..ef9508a20f 100644
--- a/samza-test/src/test/java/org/apache/samza/test/framework/AsyncStreamTaskIntegrationTest.java
+++ b/samza-test/src/test/java/org/apache/samza/test/framework/AsyncStreamTaskIntegrationTest.java
@@ -161,6 +161,4 @@ public void testSamzaJobTimeoutFailureForAsyncTask() {
         .addOutputStream(imod, 1)
         .run(Duration.ofMillis(1));
   }
-
-
 }
diff --git a/samza-test/src/test/java/org/apache/samza/test/framework/StreamTaskIntegrationTest.java b/samza-test/src/test/java/org/apache/samza/test/framework/StreamTaskIntegrationTest.java
index 0580598652..bc5cba7144 100644
--- a/samza-test/src/test/java/org/apache/samza/test/framework/StreamTaskIntegrationTest.java
+++ b/samza-test/src/test/java/org/apache/samza/test/framework/StreamTaskIntegrationTest.java
@@ -167,4 +167,4 @@ public void genData(Map<Integer, List<KV>> inputPartitionData, Map<Integer, List
       expectedOutputPartitionData.put(i, new ArrayList<Integer>(outputPartition));
     }
   }
-}
+}
\ No newline at end of file