Specify target type in generateDecodeInt()

1 files changed, 8 insertions, 3 deletions

diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatransient0/JavaTransient0Generator.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatransient0/JavaTransient0Generator.java
index 22ad5ae..b18c30a 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatransient0/JavaTransient0Generator.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatransient0/JavaTransient0Generator.java
@@ -257,7 +257,7 @@ public class JavaTransient0Generator implements Generator {
     private static Stream<JavaBlockStatement> generateDecodeInt32(
         JavaIdentifier target
     ) {
-        return generateDecodeInt(target, 32);
+        return generateDecodeInt(target, 32, JavaTypeRef.INT);
     }
 
     private static Stream<JavaBlockStatement> generateEncodeInt(
@@ -284,11 +284,16 @@ public class JavaTransient0Generator implements Generator {
 
     private static Stream<JavaBlockStatement> generateDecodeInt(
         JavaIdentifier target,
-        int bits
+        int bits,
+        JavaTypeRef javaType
     ) {
         var statements = new ArrayList<JavaBlockStatement>();
 
-        statements.add(new JavaLocalVariableDeclaration(Optional.empty(), target, new JavaIntegerLiteral(0)));
+        statements.add(new JavaLocalVariableDeclaration(
+            Optional.of(javaType),
+            target,
+            new JavaIntegerLiteral(0)
+        ));
 
         Iterable<Integer> shiftBys = () -> byteIndicesInBits(bits)
             .iterator();