fix segfault
fix a bug