1 files changed, 7 insertions, 8 deletions
diff --git a/src/main/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceRunner.scala b/src/main/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceRunner.scala
index bbcf96d..12f1982 100644
--- a/src/main/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceRunner.scala
+++ b/src/main/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceRunner.scala
@@ -10,17 +10,16 @@ object SimpleHBaseSourceRunner extends App {
 
   val log = LoggerFactory.getLogger(this.getClass.getName)
 
-
-
   log.info("Starting HBaseSource Import Process Test...")
 
   val start1 = System.currentTimeMillis
 
-  JobRunner.main((classOf[SimpleHBaseSourceExample].getName :: mArgs.toList).toArray)
-
-  val end = System.currentTimeMillis
+  try {
+    JobRunner.main((classOf[SimpleHBaseSourceExample].getName :: mArgs.toList).toArray)
+  } finally {
+    val end = System.currentTimeMillis
 
-  log.info("HBaseSource Import process finished successfully.")
-  log.info("HBaseSource Import process : " + (end - start1) + " milliseconds to complete")
-  
+    log.info("HBaseSource Import process finished successfully.")
+    log.info("HBaseSource Import process : " + (end - start1) + " milliseconds to complete")
+  }
 }
 \ No newline at end of file