From d483dfa06850b056f5b87d263b1db21be60968cd Mon Sep 17 00:00:00 2001
From: alemidev <me@alemi.dev>
Date: Tue, 4 Jan 2022 15:37:43 +0100
Subject: [PATCH] fixed type hints

---
 compiler/proto.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/compiler/proto.py b/compiler/proto.py
index ab18d10..4c3d55a 100644
--- a/compiler/proto.py
+++ b/compiler/proto.py
@@ -68,6 +68,7 @@ TYPE_MAP = {
 	"UUID": Ref('UUID'),
 	"string": Ref('String'),
 	"nbt": Ref('NBTTag'),
+	"optionalNbt": Ref('OptionalType(NBTTag)'),
 	"slot": Ref('Slot'),
 	"position": Ref('Position'),
 	"entityMetadataItem": Ref('EntityMetadataItem'),
@@ -93,8 +94,7 @@ HINT_MAP = {
 	"nbt": 'bytes',
 	"slot": 'dict',
 	"position": 'tuple',
-	"entityMetadataItem": 'bytes',
-	"entityMetadata": 'bytes',
+	"entityMetadata": 'dict',
 }
 
 def _format_line(i, depth:int=0) -> str: