2 files changed, 7 insertions, 2 deletions
diff --git a/src/main/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExample.scala b/src/main/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExample.scala
index 7ba2788..2921df8 100644
--- a/src/main/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExample.scala
+++ b/src/main/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExample.scala
@@ -6,6 +6,8 @@ import org.apache.log4j.{Level, Logger}
 import parallelai.spyglass.hbase.{HBasePipeConversions, HBaseSource}
 import parallelai.spyglass.hbase.HBaseConstants.SourceMode
 import cascading.tuple.Fields
+import cascading.property.AppProps
+import java.util.Properties
 
 /**
   * Simple example of HBaseSource usage
@@ -17,7 +19,10 @@ class SimpleHBaseSourceExample(args: Args) extends JobBase(args) with HBasePipeC
    if (isDebug) Logger.getRootLogger.setLevel(Level.DEBUG)
 
    val output = args("output")
-
+   
+   val properties = new Properties()
+   AppProps.setApplicationJarClass( properties, classOf[SimpleHBaseSourceExample] );
+   
    val hbs = new HBaseSource(
      "table_name",
      "quorum_name:2181",
diff --git a/src/test/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExampleTest.scala b/src/test/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExampleTest.scala
index 93ee129..f25e769 100644
--- a/src/test/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExampleTest.scala
+++ b/src/test/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExampleTest.scala
@@ -30,7 +30,7 @@ class SimpleHBaseSourceExampleTest extends FunSpec with TupleConversions {
   )
 
   JobTest("parallelai.spyglass.hbase.example.SimpleHBaseSourceExample")
-    .arg("local", "")
+    .arg("test", "")
     .arg("app.conf.path", "app.conf")
     .arg("output", output)
     .arg("debug", "true")