genblocks

1 files changed, 5 insertions, 2 deletions

diff --git a/codegen/newpacket.py b/codegen/newpacket.py
index 52d6a2b4..48d97640 100644
--- a/codegen/newpacket.py
+++ b/codegen/newpacket.py
@@ -1,11 +1,14 @@
+import lib.code.version
 import lib.code.packet
 import lib.code.utils
 import lib.download
 import lib.extract
 import sys
 
-mappings = lib.download.get_mappings_for_version('1.18.2')
-burger_data = lib.extract.get_burger_data_for_version('1.18.2')
+version_id = lib.code.version.get_version_id()
+
+mappings = lib.download.get_mappings_for_version(version_id)
+burger_data = lib.extract.get_burger_data_for_version(version_id)
 
 burger_packets_data = burger_data[0]['packets']['packet']
 packet_id, direction, state = int(sys.argv[1]), sys.argv[2], sys.argv[3]