Memary leak fix