fixed a segfault bug