fix possible segfault
fix possible segfault