Fix possible segfault