diff --git a/package-lock.json b/package-lock.json
index e5aab2d..b3ba777 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "@solarity/zktype",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "@solarity/zktype",
-      "version": "0.3.0",
+      "version": "0.3.1",
       "license": "MIT",
       "dependencies": {
         "ejs": "3.1.10",
diff --git a/package.json b/package.json
index 1e1d858..c3e7aec 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@solarity/zktype",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "description": "Unleash TypeScript bindings for Circom circuits",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
diff --git a/src/core/templates/utils.ts b/src/core/templates/utils.ts
index edd1fdb..d401314 100644
--- a/src/core/templates/utils.ts
+++ b/src/core/templates/utils.ts
@@ -7,8 +7,8 @@ export function normalizePublicSignals(
 ): any {
   let index = 0;
   return signalNames.reduce((acc: any, signalName) => {
-    const dimensions = getSignalDimensions(signalName);
-    const size = dimensions.reduce((a, b) => a * b, 1);
+    const dimensions: number[] = getSignalDimensions(signalName);
+    const size: number = dimensions.reduce((a, b) => a * b, 1);
 
     acc[signalName] = reshape(publicSignals.slice(index, index + size), dimensions);
     index += size;
@@ -29,9 +29,9 @@ function reshape(array: number[], dimensions: number[]): any {
   }
 
   const [first, ...rest] = dimensions;
-  const size = rest.reduce((a, b) => a * b, 1);
+  const size: number = rest.reduce((a, b) => a * b, 1);
 
-  const result = [];
+  const result: any[] = [];
   for (let i = 0; i < first; i++) {
     result.push(reshape(array.slice(i * size, (i + 1) * size), rest));
   }