1 files changed, 5 insertions, 6 deletions

diff --git a/src/main/scala/uk/ac/ox/cs/rsacomb/Main.scala b/src/main/scala/uk/ac/ox/cs/rsacomb/Main.scala
index 7eb39e8..fe88b4f 100644
--- a/src/main/scala/uk/ac/ox/cs/rsacomb/Main.scala
+++ b/src/main/scala/uk/ac/ox/cs/rsacomb/Main.scala
@@ -41,6 +41,10 @@ object RSAComb extends App {
   if (config.contains('logger))
     Logger.level = config('logger).get[Logger.Level]
 
+  /* Set answers output file */
+  if (config.contains('answers))
+    Logger.answers = config('answers).get[os.Path]
+
   /* Load original ontology and normalize it */
   val ontopath = config('ontology).get[os.Path]
   val data = config('data).get[List[os.Path]]
@@ -61,12 +65,7 @@ object RSAComb extends App {
     val answers = rsa ask queries
 
     /* Write answers to output file */
-    os.write(
-      config('answers).get[os.Path],
-      ujson.write(ujson.Arr(answers.map(_.toJSON)), indent = 2),
-      createFolders = true
-    )
-
+    Logger write answers
     /* Generate simulation script */
     Logger.generateSimulationScripts(data, queries)
   }