Found out issue, trying to fix alloc