Fixed a segfault....