segfault bug fixed