From 1782863dbcdfb64b30f53f402456302fd99411c4 Mon Sep 17 00:00:00 2001
From: Jorel Ali <github@jorel.dev>
Date: Tue, 12 Dec 2023 16:59:07 +0000
Subject: [PATCH] Fix invalid generated code from NBT Compound argument in
 annotation example project

---
 .../dev/jorel/commandapi/annotations/Annotations.java     | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/commandapi-annotations/src/main/java/dev/jorel/commandapi/annotations/Annotations.java b/commandapi-annotations/src/main/java/dev/jorel/commandapi/annotations/Annotations.java
index 5facf40a24..0880d6497d 100644
--- a/commandapi-annotations/src/main/java/dev/jorel/commandapi/annotations/Annotations.java
+++ b/commandapi-annotations/src/main/java/dev/jorel/commandapi/annotations/Annotations.java
@@ -305,8 +305,12 @@ private <T extends Annotation> Map<Integer, String> emitArgumentsAndGenerateArgu
 			
 			// Handle return types
 			Primitive primitive = getPrimitive(argumentAnnotation);
-			if(primitive.value().length == 1) {
-				argumentMapping.put(i - 1, primitive.value()[0]);
+			if (argumentAnnotation instanceof ANBTCompoundArgument) {
+				argumentMapping.put(i - 1, parameter.asType().toString());
+			} else {
+				if(primitive.value().length == 1) {
+					argumentMapping.put(i - 1, primitive.value()[0]);
+				}
 			}
 		}