Fixed segfault (oops)