Fix the crash error