Fixed segfault....