1 files changed, 3 insertions, 4 deletions
diff --git a/src/main/scala/org/perl8/test/tap/TestBuilder.scala b/src/main/scala/org/perl8/test/tap/TestBuilder.scala
index 203080c..b1d8579 100644
--- a/src/main/scala/org/perl8/test/tap/TestBuilder.scala
+++ b/src/main/scala/org/perl8/test/tap/TestBuilder.scala
@@ -51,13 +51,13 @@ class TestBuilder (
     throw new BailOutException(message.getOrElse(""))
   }
 
-  def doneTesting () {
+  def doneTesting (): Boolean = {
     plan match {
       case None => outLine(Producer.plan(state.currentTest - 1))
       case _    => ()
     }
 
-    if (!isPassing) {
+    if (!state.isPassing) {
       if (!state.matchesPlan) {
         val planCount = (plan match {
           case Some(p) => p.plan
@@ -80,10 +80,9 @@ class TestBuilder (
         diag("Looks like you failed " + fails + " of " + total + ".")
       }
     }
-  }
 
-  def isPassing: Boolean =
     state.isPassing
+  }
 
   def failedTests: Int =
     state.failCount