Possible segfault fix