Fixed a segfault error