From ed56ef17a1f6bef78428b047e8bb320bd9605f43 Mon Sep 17 00:00:00 2001
From: Martin Sumner <martin.sumner@adaptip.co.uk>
Date: Wed, 13 Sep 2017 23:45:48 +0100
Subject: [PATCH] Make export mochijson friendly

---
 src/leveled_tictac.erl | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/leveled_tictac.erl b/src/leveled_tictac.erl
index 5e9fe0a..ad420cc 100644
--- a/src/leveled_tictac.erl
+++ b/src/leveled_tictac.erl
@@ -121,7 +121,8 @@ new_tree(TreeID, Size) ->
 export_tree(Tree) ->
     L2 = 
         lists:foldl(fun(X, L2Acc) ->
-                            [{X, array:get(X, Tree#tictactree.level2)}|L2Acc]
+                            [{integer_to_binary(X), 
+                                array:get(X, Tree#tictactree.level2)}|L2Acc]
                         end,
                     [],
                     lists:seq(0, Tree#tictactree.width - 1)),
@@ -146,11 +147,11 @@ import_tree(ExportedTree) ->
     {Size, Width} = lists:keyfind(Width, 2, Sizes),
     {BitWidth, Width, SegmentCount} = get_size(Size),
     Lv2Init = array:new([{size, Width}]),
-    Lv2 = lists:foldl(fun({X, L2SegBin}, L2Array) ->
-                                array:set(X, L2SegBin, L2Array)
-                            end,
-                        Lv2Init,
-                        L2List),
+    FoldFun = 
+        fun({X, L2SegBin}, L2Array) ->
+            array:set(binary_to_integer(X), L2SegBin, L2Array)
+        end,
+    Lv2 = lists:foldl(FoldFun, Lv2Init, L2List),
     #tictactree{treeID = import,
                     size = Size,
                     width = Width,