Fix seg fault bug.