Fix some crash