Fixed the segfault :)