Fixed segfault error