fixed segfault bug