Probable segfault fix