diff --git a/swl_pack.py b/swl_pack.py
index c5f0d02..a9722c8 100644
--- a/swl_pack.py
+++ b/swl_pack.py
@@ -1,5 +1,5 @@
 import sys, json
-from pydofus.swl import SWLReader, InvalidSWLFile
+from pydofus.swl import SWLBuilder
 
 # python swl_pack.py file.swf (require file.json)
 # file output: file.swl
diff --git a/swl_unpack.py b/swl_unpack.py
index 2801b96..47ec8b3 100644
--- a/swl_unpack.py
+++ b/swl_unpack.py
@@ -12,7 +12,7 @@
 
 swl = SWLReader(swl_input)
 swf_output.write(swl.SWF)
-swl_data = {'version':swl_reader.version, 'frame_rate':swl_reader.frame_rate, 'classes':swl_reader.classes}
+swl_data = {'version': swl.version, 'frame_rate': swl.frame_rate, 'classes': swl.classes}
 json.dump(swl_data, json_output, indent=4)
 
 swl_input.close()