Tidyup and fix case where definition does not exist

1 files changed, 2 insertions, 4 deletions

diff --git a/ybd.py b/ybd.py
index 40cf259..feeb684 100755
--- a/ybd.py
+++ b/ybd.py
@@ -31,9 +31,7 @@ target = os.path.splitext(os.path.basename(sys.argv[1]))[0]
 arch = sys.argv[2]
 with app.setup(target, arch):
     with app.timer('TOTAL', 'YBD starts'):
-        defs = Definitions()
-        definition = defs.get(target)
         with app.timer('CACHE-KEYS', 'Calculating'):
             cache.get_cache(target)
-        defs.save_trees()
-        assemble(definition)
+        Definitions().save_trees()
+        assemble(target)