1 files changed, 5 insertions, 6 deletions
diff --git a/mavlink-library/src/test/scala/com/github/jodersky/mavlink/MainTest.scala b/mavlink-library/src/test/scala/com/github/jodersky/mavlink/MainTest.scala
index faf252a..be46212 100644
--- a/mavlink-library/src/test/scala/com/github/jodersky/mavlink/MainTest.scala
+++ b/mavlink-library/src/test/scala/com/github/jodersky/mavlink/MainTest.scala
@@ -1,16 +1,15 @@
 package com.github.jodersky.mavlink
 
-import scala.io.Source
-import scala.xml.XML
-import trees._
+import java.io.File
 
 object MainTest {
   
   def main(args: Array[String]): Unit = {
-    val definition = XML.load(getClass.getResource("/concise.xml"))
-    val dialect = Parser.parseDialect(definition)
+    val parser: Parser = new Parser(StandardReporter)
+    val dialectDefinitionFile: File = new File("mavlink-library/src/test/resources/including.xml")
+    val dialect = parser.parseDialect(dialectDefinitionFile)
     val generator = new Generator(dialect)
-    println(generator.generate())
+    println(generator.targets.map(_.generate()).mkString("\n\n"))
   }
 
 }
 \ No newline at end of file