diff --git a/lib/array.cr b/lib/array.cr
index 810250a..e703599 100644
--- a/lib/array.cr
+++ b/lib/array.cr
@@ -127,11 +127,11 @@ class Array(T)
   end
 end
 
-def get_shape(arr : Array(Array(T)))
+def get_shape(arr : Array(Array(T))) forall T
   {arr.size, arr[0].size}
 end
 
-def get_shape(arr : Array(T))
+def get_shape(arr : Array(T)) forall T
   arr.size
 end