Fix OOM bug