Fixed problem with memory allocation