Fix alloc failure handling