Handle prototypes a bit better, but I suspect not everywhere

1 files changed, 8 insertions, 4 deletions

diff --git a/memory.c b/memory.c
index f1ff3a3..c94993c 100644
--- a/memory.c
+++ b/memory.c
@@ -64,11 +64,15 @@ freearray(Array *a)
 
 	SetRefs(a, GetRefs(a)-1);
 	if(GetRefs(a) == 0){
-		if(GetType(a) == AtypeArray)
-			for(int i = 0; i < GetSize(a); i++)
-				freearray(a->arraydata[i]);
+		if(GetType(a) == AtypeArray){
+			if(GetSize(a) > 0){
+				for(int i = 0; i < GetSize(a); i++)
+					freearray(a->arraydata[i]);
+				free(a->rawdata);
+			}else
+				freearray(a->prototype);
+		}
 		free(a->shape);
-		free(a->rawdata);
 		free(a);
 		arrayalloccounts--;
 	}